wizata-dsapi 2.1.0.dev7__tar.gz → 2.1.0.dev9__tar.gz

{wizata_dsapi-2.1.0.dev7 → wizata_dsapi-2.1.0.dev9}/PKG-INFO

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

{wizata_dsapi-2.1.0.dev7 → wizata_dsapi-2.1.0.dev9}/wizata_dsapi/context.py

@@ -68,6 +68,13 @@ class Context(ILogger):
         self.pipeline_id = pipeline_id
         self.execution_id = execution_id
         self.experiment_id = experiment_id
+        # Git ref the pipeline body was loaded at (DEV-5895): sha is the exact
+        # commit that ran (stamped on the ExecutionLog, DEV-5950), branch is the
+        # targeted branch when any. RAW-script resolution reads scripts/<name>.py
+        # from git at this sha so a branch run executes the branch's script
+        # version — not the Postgres "current" one. Left None off the git path.
+        self.branch = None
+        self.sha = None
         self.template = template
         self.registration = registration
         self.now = now

wizata_dsapi-2.1.0.dev9/wizata_dsapi/version.py

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

{wizata_dsapi-2.1.0.dev7 → wizata_dsapi-2.1.0.dev9}/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.dev7 → wizata_dsapi-2.1.0.dev9}/wizata_dsapi.egg-info/PKG-INFO

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

wizata_dsapi-2.1.0.dev7/wizata_dsapi/version.py

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