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

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

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

wizata_dsapi-2.1.0.dev8/wizata_dsapi/version.py

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

{wizata_dsapi-2.1.0.dev7 → 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.dev7 → 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.dev7
+Version: 2.1.0.dev8
 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"