wizata-dsapi 2.1.0.dev6__tar.gz → 2.1.0.dev8__tar.gz

{wizata_dsapi-2.1.0.dev6 → wizata_dsapi-2.1.0.dev8}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: wizata_dsapi
-Version: 2.1.0.dev6
+Version: 2.1.0.dev8
 Summary: Wizata Data Science Toolkit
 Author: Wizata S.A.
 Author-email: info@wizata.com

{wizata_dsapi-2.1.0.dev6 → wizata_dsapi-2.1.0.dev8}/wizata_dsapi/execution_log.py

@@ -79,7 +79,9 @@ class ExecutionLog(Dto):
             message=data.get("message"),
             messages=data.get("messages"),
             timestamp=int(data["timestamp"]) if "timestamp" in data and data["timestamp"] is not None else None,
-            level=data.get("level")
+            level=data.get("level"),
+            branch=data.get("branch"),
+            sha=data.get("sha"),
         )
 
     def __init__(self,
@@ -104,6 +106,8 @@ class ExecutionLog(Dto):
                  message: str = None,
                  messages: list = None,
                  level: str = None,
+                 branch: str = None,
+                 sha: str = None,
                  ):
 
         # labels
@@ -134,12 +138,18 @@ class ExecutionLog(Dto):
         self.edge_device_id = edge_device_id
         self.status = status
         self.level = level
+        # git branch this run TARGETS (which version to execute) — a label so
+        # runs are queryable per branch. None = default branch / non-git.
+        self.branch = branch
 
         # values
         if execution_id is None:
             execution_id = uuid.uuid4()
         self.execution_id = execution_id
         self.pipeline_image_id = pipeline_image_id
+        # resolved commit SHA the run executed — INFO only, from the branch HEAD
+        # (git) or the image id. Not a target.
+        self.sha = sha
         self.timestamp = timestamp
         self.started_date = started_date
         self.queued_date = queued_date
@@ -428,7 +438,8 @@ class ExecutionLog(Dto):
             "twin": self.get_twin_key(),
             "edgeDeviceId": self.edge_device_id,
             "status": self.status.value if hasattr(self.status, 'value') else self.status,
-            "level": self.level
+            "level": self.level,
+            "branch": self.branch,
         }
         return {k: v for k, v in labels.items() if v is not None}
 
@@ -443,6 +454,7 @@ class ExecutionLog(Dto):
             "executionTime": self.execution_time,
             "executionTriggerId": str(self.trigger_id) if self.trigger_id else None,
             "version": self.version,
+            "sha": self.sha,
             "message": self.message,
             "messages": self.messages or None
         }

wizata_dsapi-2.1.0.dev8/wizata_dsapi/version.py

@@ -0,0 +1 @@
+__version__ = "2.1.0.dev8"

{wizata_dsapi-2.1.0.dev6 → wizata_dsapi-2.1.0.dev8}/wizata_dsapi/wizata_dsapi_client.py

@@ -929,6 +929,7 @@ class WizataDSAPIClient(ApiInterface, ApiDtoInterface):
                   twin=None,
                   mode: str = 'experiment',
                   version: str = None,
+                  branch: str = None,
                   ) -> ExecutionLog:
         """
         wrapper to execute a pipeline, used by other functions.
@@ -941,7 +942,8 @@ class WizataDSAPIClient(ApiInterface, ApiDtoInterface):
                 pipeline=pipeline,
                 twin=twin,
                 pipeline_image_id=image,
-                version=version
+                version=version,
+                branch=branch
             )
 
         if isinstance(execution_log, ExecutionLog):
@@ -968,7 +970,8 @@ class WizataDSAPIClient(ApiInterface, ApiDtoInterface):
                    train: bool = True,
                    plot: bool = True,
                    write: bool = False,
-                   version: str = None) -> ExecutionLog:
+                   version: str = None,
+                   branch: str = None) -> ExecutionLog:
         """
         experiment and train models with a pipeline.
 
@@ -988,6 +991,7 @@ class WizataDSAPIClient(ApiInterface, ApiDtoInterface):
         :param plot: if False plot steps are ignored.
         :param write: if False write steps are ignored.
         :param version: version to use, by default use your python environment version.
+        :param branch: git branch to run against (git mode); default branch when omitted.
         """
         if not train or not plot or write:
             if properties is None:
@@ -1005,7 +1009,8 @@ class WizataDSAPIClient(ApiInterface, ApiDtoInterface):
             image=image,
             properties=properties,
             mode='experiment',
-            version=version
+            version=version,
+            branch=branch
         )
 
     def run(self,
@@ -1015,7 +1020,8 @@ class WizataDSAPIClient(ApiInterface, ApiDtoInterface):
             train: bool = False,
             plot: bool = False,
             write: bool = True,
-            version: str = None) -> ExecutionLog:
+            version: str = None,
+            branch: str = None) -> ExecutionLog:
         """
         run a pipeline.
 
@@ -1035,6 +1041,7 @@ class WizataDSAPIClient(ApiInterface, ApiDtoInterface):
         :param plot: if False plot steps are ignored.
         :param write: if False write steps are ignored.
         :param version: version of python environment to use if different than local environment.
+        :param branch: git branch to run against (git mode); default branch when omitted.
         """
 
         if train or plot or not write:
@@ -1052,7 +1059,8 @@ class WizataDSAPIClient(ApiInterface, ApiDtoInterface):
             properties=properties,
             image=image,
             mode='production',
-            version=version
+            version=version,
+            branch=branch
         )
 
     def await_execution(self,

{wizata_dsapi-2.1.0.dev6 → wizata_dsapi-2.1.0.dev8}/wizata_dsapi.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: wizata_dsapi
-Version: 2.1.0.dev6
+Version: 2.1.0.dev8
 Summary: Wizata Data Science Toolkit
 Author: Wizata S.A.
 Author-email: info@wizata.com

wizata_dsapi-2.1.0.dev6/wizata_dsapi/version.py

@@ -1 +0,0 @@
-__version__ = "2.1.0.dev6"