code-loader 1.0.47__tar.gz → 1.0.48__tar.gz

{code_loader-1.0.47 → code_loader-1.0.48}/PKG-INFO

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

{code_loader-1.0.47 → code_loader-1.0.48}/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-1.0.47 → code_loader-1.0.48}/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-1.0.47 → code_loader-1.0.48}/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.47 → code_loader-1.0.48}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "code-loader"
-version = "1.0.47"
+version = "1.0.48"
 description = ""
 authors = ["dorhar <doron.harnoy@tensorleap.ai>"]
 license = "MIT"