code-loader 1.0.46__py3-none-any.whl → 1.0.48__py3-none-any.whl

code_loader/experiment_api/api.py

@@ -42,7 +42,7 @@ class LogExternalEpochDataRequest:
   experimentId: str
   epoch: int
   metrics: ApiMetrics
-  force: bool = False
+  override: bool = False
 
 @dataclass
 class TagModelRequest:

code_loader/experiment_api/epoch.py

@@ -91,13 +91,14 @@ class Epoch:
             tags=tags
         ))
 
-    def log(self, modelFilePath: Optional[str] = None, tags: Optional[list[str]] = None) -> None:
+    def log(self, modelFilePath: Optional[str] = None, tags: Optional[List[str]] = None, override: bool = False) -> None:
         """
         Logs the epoch with optional model file and tags.
 
         Args:
             modelFilePath (Optional[str]): The path to the model file. Defaults to None.
             tags (Optional[List[str]]): A list of tags to associate with the epoch model. Will always include the default tag. Unless the default tag is set to '', all previous epoch model with the same tag will be removed
+            override (bool): Whether to override the existing epoch model. Defaults to False.
         """
         if tags is None:
             tags = []
@@ -119,7 +120,8 @@ class Epoch:
             experimentId=self.ctx.experiment_id,
             projectId=self.ctx.project_id,
             epoch=self.epoch,
-            metrics=api_metrics
+            metrics=api_metrics,
+            override=override
         ))
         if modelFilePath is not None and len(tags) > 0:
             self._tag_model(tags)

code_loader/experiment_api/experiment.py

@@ -41,7 +41,7 @@ class Experiment:
         """
         return Epoch(self.ctx, epoch, self.default_epoch_tag)
     
-    def log_epoch(self, epoch: int, metrics: Optional[Metrics] = None, model_path: Optional[str] = None, tags: Optional[List[str]] = None) -> None:
+    def log_epoch(self, epoch: int, metrics: Optional[Metrics] = None, model_path: Optional[str] = None, tags: Optional[List[str]] = None, override: bool = False) -> None:
         """
         Logs an epoch with optional metrics, model path, and tags.
 
@@ -50,11 +50,12 @@ class Experiment:
             metrics (Optional[Metrics]): The metrics to log for the epoch. Defaults to None.
             model_path (Optional[str]): The path to the model file. Defaults to None.
             tags (Optional[List[str]]): A list of tags to associate with the epoch model. Will always include the default tag. Unless the default tag is set to '', all previous epoch model with the same tag will be removed.
+            override (bool): Whether to override the epoch if it already exists. Defaults to False.
         """
         epoch_o = self.init_epoch(epoch)
         if metrics is not None:
             epoch_o.set_metrics(metrics)
-        epoch_o.log(model_path, tags)
+        epoch_o.log(model_path, tags, override)
 
     def set_properties(self, properties: Dict[str, Any]) -> None:
         """

{code_loader-1.0.46.dist-info → code_loader-1.0.48.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: code-loader
-Version: 1.0.46
+Version: 1.0.48
 Summary: 
 Home-page: https://github.com/tensorleap/code-loader
 License: MIT

{code_loader-1.0.46.dist-info → code_loader-1.0.48.dist-info}/RECORD

@@ -7,11 +7,11 @@ code_loader/contract/exceptions.py,sha256=jWqu5i7t-0IG0jGRsKF4DjJdrsdpJjIYpUkN1F
 code_loader/contract/responsedataclasses.py,sha256=w7xVOv2S8Hyb5lqyomMGiKAWXDTSOG-FX1YW39bXD3A,3969
 code_loader/contract/visualizer_classes.py,sha256=Ka8fJSVKrOeZ12Eg8-dOBGMW0UswYCIkuhnNMd-7z9s,22948
 code_loader/experiment_api/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-code_loader/experiment_api/api.py,sha256=p_III5t9WCpvd1REq9L-pF3r1RhjDlxMCBBK-Sn0B1I,2490
+code_loader/experiment_api/api.py,sha256=a7wh6Hhe7IaVxu46eV2soSz-yxnmXG3ipU1BBtsEAaQ,2493
 code_loader/experiment_api/cli_config_utils.py,sha256=n6JMyNrquxql3KKxHhAP8jAzezlRT-PV2KWI95kKsm0,1140
 code_loader/experiment_api/client.py,sha256=8kqLWh2Zg7mNA5E9Q0v0qRuPm3OE3Rs7lfGAcXOucU4,1428
-code_loader/experiment_api/epoch.py,sha256=OwK8ENR7TJIkreUAZKTRJamPcHDvQYeU85qcDNxSCpo,4700
-code_loader/experiment_api/experiment.py,sha256=oVi6GAJ1e1kOygnOquTvb6SgepMhTlrpH0uVqCwviYM,6349
+code_loader/experiment_api/epoch.py,sha256=mcrdgwbUgG8hcq6Nthp7bNl5KAVZ5727_ICasJp7thQ,4849
+code_loader/experiment_api/experiment.py,sha256=vyE2zyY9JcVKuHcIA8Njr5DWlmbP2cyJHnGAsnaOots,6483
 code_loader/experiment_api/experiment_context.py,sha256=kdzUbuzXo1pMVslOC3TKeJwW8sx_qWkxDVrswjduH0A,194
 code_loader/experiment_api/types.py,sha256=MY8xFARHwdVA7p4dxyhD60ShmttgTvb4qdp1oEB_NPg,485
 code_loader/experiment_api/utils.py,sha256=XZHtxge12TS4H4-8PjV3sKuhp8Ud6ojAiIzTZJEqBqc,3304
@@ -22,7 +22,7 @@ code_loader/leaploader.py,sha256=E1DHtRDzz0hyb3kAMl6S_JDbjGxJPR_bZHU77Af4rR4,175
 code_loader/utils.py,sha256=TZAoUbA2pE8eK3Le3s5Xr4eRaYdeDMQtxotx6rh-5oE,2185
 code_loader/visualizers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 code_loader/visualizers/default_visualizers.py,sha256=VoqO9FN84yXyMjRjHjUTOt2GdTkJRMbHbXJ1cJkREkk,2230
-code_loader-1.0.46.dist-info/LICENSE,sha256=qIwWjdspQeSMTtnFZBC8MuT-95L02FPvzRUdWFxrwJY,1067
-code_loader-1.0.46.dist-info/METADATA,sha256=25rGAJqHthPl5csB16XuFX9HkycPAjnmO-5Mo35SI0U,888
-code_loader-1.0.46.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-code_loader-1.0.46.dist-info/RECORD,,
+code_loader-1.0.48.dist-info/LICENSE,sha256=qIwWjdspQeSMTtnFZBC8MuT-95L02FPvzRUdWFxrwJY,1067
+code_loader-1.0.48.dist-info/METADATA,sha256=s65JNpeKwOhpvCS8TY1RFkfSEUTDxGCfiBBjy-b7sh8,888
+code_loader-1.0.48.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+code_loader-1.0.48.dist-info/RECORD,,