wizata-dsapi 1.3.10__py3-none-any.whl → 1.3.12__py3-none-any.whl

wizata_dsapi/mlmodel.py

@@ -24,6 +24,7 @@ class MLModelConfig(ApiDto):
     :ivar str model_key: key of the model to store (or use story property if dynamic).
     :ivar str model_type: reserved for 'mlflow' defines model flavor such as 'sklearn' or 'pyfunc'.
     :ivar str model_alias: reserved for 'mlflow' set the alias to target and use inside a pipeline.
+    :ivar str model_format: type of model (by default 'pkl') accept also 'pt' for PyTorch model using tensors (none = 'pkl')
     :ivar dict properties_mapping: dict to map properties expected by the script with the properties from the context.
     :ivar str property_name: define which property is looked for when using 'by_property'.
     :ivar str source: define name of model storage to use 'wizata'.
@@ -40,7 +41,6 @@ class MLModelConfig(ApiDto):
                  train_script=None,
                  train_test_split_pct: float = 1.0,
                  train_test_split_type: str = "ignore",
-                 format: str = 'pkl',
                  function: str = "predict",
                  properties_mapping=None,
                  target_feat=None,
@@ -51,6 +51,7 @@ class MLModelConfig(ApiDto):
                  model_key: str = None,
                  model_type: str = None,
                  model_alias: str = None,
+                 model_format: str = 'pkl',
                  by_twin: bool = False,
                  by_property: bool = False,
                  property_name: str = None,
@@ -76,7 +77,7 @@ class MLModelConfig(ApiDto):
         self.by_twin = by_twin
         self.by_property = by_property
         self.property_name = property_name
-        self.format = format
+        self.model_format = model_format
 
         # source
         self.source = source
@@ -159,8 +160,8 @@ class MLModelConfig(ApiDto):
             self.model_type = obj["model_type"]
         if "model_alias" in obj.keys() and obj["model_alias"] is not None:
             self.model_alias = obj["model_alias"]
-        if "format" in obj.keys() and obj["format"] is not None:
-            self.format = obj["format"]
+        if "model_format" in obj.keys() and obj["model_format"] is not None:
+            self.model_format = obj["model_format"]
 
     def to_json(self, target: str = None):
         obj = {
@@ -204,8 +205,8 @@ class MLModelConfig(ApiDto):
             obj["by_property"] = str(self.by_property)
         if self.property_name is not None:
             obj["property_name"] = self.property_name
-        if self.format  is not None:
-            obj["format"] = self.format
+        if self.model_format  is not None:
+            obj["model_format"] = self.model_format
 
         return obj
 

wizata_dsapi/version.py

@@ -1 +1 @@
-__version__ = "1.3.10"
+__version__ = "1.3.12"

{wizata_dsapi-1.3.10.dist-info → wizata_dsapi-1.3.12.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: wizata_dsapi
-Version: 1.3.10
+Version: 1.3.12
 Summary: Wizata Data Science Toolkit
 Author: Wizata S.A.
 Author-email: info@wizata.com
@@ -12,8 +12,9 @@ Requires-Dist: numpy==1.26.4; python_version < "3.12"
 Requires-Dist: numpy>=2.3.3; python_version >= "3.12"
 Requires-Dist: dill==0.3.6; python_version < "3.12"
 Requires-Dist: dill==0.4.0; python_version >= "3.12"
-Requires-Dist: protobuf>=3.20.3
-Requires-Dist: msal==1.24.0; python_version < "3.12"
+Requires-Dist: protobuf==5.25.9; python_version < "3.12"
+Requires-Dist: protobuf>=5.25.9; python_version >= "3.12"
+Requires-Dist: msal==1.30.0; python_version < "3.12"
 Requires-Dist: msal>=1.34.0; python_version >= "3.12"
 Requires-Dist: joblib==1.5.2; python_version < "3.12"
 Requires-Dist: joblib>=1.2.0; python_version >= "3.12"
@@ -31,8 +32,9 @@ Requires-Dist: numpy==1.26.4; python_version < "3.12" and extra == "all"
 Requires-Dist: numpy>=2.3.3; python_version >= "3.12" and extra == "all"
 Requires-Dist: dill==0.3.6; python_version < "3.12" and extra == "all"
 Requires-Dist: dill==0.4.0; python_version >= "3.12" and extra == "all"
-Requires-Dist: protobuf>=3.20.3; extra == "all"
-Requires-Dist: msal==1.24.0; python_version < "3.12" and extra == "all"
+Requires-Dist: protobuf==5.25.9; python_version < "3.12" and extra == "all"
+Requires-Dist: protobuf>=5.25.9; python_version >= "3.12" and extra == "all"
+Requires-Dist: msal==1.30.0; python_version < "3.12" and extra == "all"
 Requires-Dist: msal>=1.34.0; python_version >= "3.12" and extra == "all"
 Requires-Dist: joblib==1.5.2; python_version < "3.12" and extra == "all"
 Requires-Dist: joblib>=1.2.0; python_version >= "3.12" and extra == "all"

{wizata_dsapi-1.3.10.dist-info → wizata_dsapi-1.3.12.dist-info}/RECORD

@@ -16,7 +16,7 @@ wizata_dsapi/experiment.py,sha256=QYQ1CJ-MTWsXq08xYbm5sAp95dRxbPOmGDgaAOoBMDQ,46
 wizata_dsapi/group_system.py,sha256=6rUKe0_J3YWACysyBlzuw_TEpKNXgLOMxhpWsNxOzwY,1708
 wizata_dsapi/ilogger.py,sha256=iYnID-Z-qrYhie26C43404aIuU4_tHSKXbDeQIdo82Q,807
 wizata_dsapi/insight.py,sha256=ABFZ04DqYxxzqAEfU1tzlTZqqrigM-zN-8Lbetko3g0,6468
-wizata_dsapi/mlmodel.py,sha256=1vd5dHob3LtrjbWjpGr90GyPEcdBKxUQRxExX1OJdWc,14923
+wizata_dsapi/mlmodel.py,sha256=VwjiG6nmljA2MOlAL04jxMBgR3FmNa6NIQWNGDlTmEQ,15108
 wizata_dsapi/model_toolkit.py,sha256=a76ckSuetSKDjEkOl7o49g9oaItdw9iOaPt0JcstmhU,1551
 wizata_dsapi/paged_query_result.py,sha256=0Iyt2Kd4tvrfthhT-tk9EmSERsbJTaPNON2euHcBn6k,1150
 wizata_dsapi/pipeline.py,sha256=WDJeOxPZJiYW1qwTNZUm3jom2epIxqrSoiUwcrTF9EE,31300
@@ -31,7 +31,7 @@ wizata_dsapi/template.py,sha256=wtCRKKk3PchH4RrNgNYlEF_9C6bzZwKIeLyEvgv6Fdo,1370
 wizata_dsapi/trigger.py,sha256=w3BZYP-L3SUwvaT0oCTanh_Ewn57peZvlt7vxzHv9J8,5129
 wizata_dsapi/twin.py,sha256=S0DUzQf1smZXZTdXpXZPtkZYCfKIhw53EecCnsl9i4Q,11017
 wizata_dsapi/twinregistration.py,sha256=Mi6-YuwroiEXc0c1hgrOaphh4hNVoHupxOnXedVtJtE,13377
-wizata_dsapi/version.py,sha256=4o4BxiWDvKULo_NByGymiLj9KXGht1PsOBGUMmasvxM,23
+wizata_dsapi/version.py,sha256=2Rgd7p7gALBiXufv4uSeaC9anR_muEfi4m5NMAgfMHI,23
 wizata_dsapi/wizard_function.py,sha256=RbM7W7Gf-6Rhp_1dU9DBYkHaciknGAGvuAndhAS_vyo,942
 wizata_dsapi/wizard_request.py,sha256=v6BaqKLKvTWmUSo0_gda9FabAQz5x_-GOH1Av50GzFo,3762
 wizata_dsapi/wizata_dsapi_client.py,sha256=kos_8j47CBXiy6y6b_H2yDUe3aWh76diEmWjCisYKkY,78662
@@ -42,8 +42,8 @@ wizata_dsapi/plots/__init__.py,sha256=qgnSFqrjOPur-807M8uh5awIfjM1ZHXUXcAqHc-r2l
 wizata_dsapi/plots/common.py,sha256=jdPsJqLHBwSKc6dX83BSGPqSRxzIVNHSYO5yI_8sjGk,6568
 wizata_dsapi/scripts/__init__.py,sha256=hAxiETSQf0qOHde1si1tEAJU48seqEgHrchCzS2-LvQ,80
 wizata_dsapi/scripts/common.py,sha256=efwq-Rd0lvYljIs3gSFz9izogBD7asOU2cTK-IvHTkM,4244
-wizata_dsapi-1.3.10.dist-info/licenses/LICENSE.txt,sha256=QwcOLU5TJoTeUhuIXzhdCEEDDvorGiC6-3YTOl4TecE,11356
-wizata_dsapi-1.3.10.dist-info/METADATA,sha256=-7gm_5NICxt9GbUSCByi43lCgE8ppfns4ESqxK7vweU,5463
-wizata_dsapi-1.3.10.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-wizata_dsapi-1.3.10.dist-info/top_level.txt,sha256=-OeTJbEnh5DuWyTOHtvw0Dw3LRg3G27TNS6W4ZtfwPs,13
-wizata_dsapi-1.3.10.dist-info/RECORD,,
+wizata_dsapi-1.3.12.dist-info/licenses/LICENSE.txt,sha256=QwcOLU5TJoTeUhuIXzhdCEEDDvorGiC6-3YTOl4TecE,11356
+wizata_dsapi-1.3.12.dist-info/METADATA,sha256=zgYHssezx3tkbD3--DWp9CbOndPXQn28moPsBwz-2rs,5651
+wizata_dsapi-1.3.12.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+wizata_dsapi-1.3.12.dist-info/top_level.txt,sha256=-OeTJbEnh5DuWyTOHtvw0Dw3LRg3G27TNS6W4ZtfwPs,13
+wizata_dsapi-1.3.12.dist-info/RECORD,,