wizata-dsapi 2.2.0.dev27__tar.gz → 2.2.0.dev28__tar.gz

{wizata_dsapi-2.2.0.dev27 → wizata_dsapi-2.2.0.dev28}/PKG-INFO

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

{wizata_dsapi-2.2.0.dev27 → wizata_dsapi-2.2.0.dev28}/wizata_dsapi/mlmodel.py

@@ -1,6 +1,7 @@
 from typing import List, Iterator, Union, Optional
 import os
 from .api_dto import ApiDto, ApiDtoInterface
+from .script import ScriptType
 from datetime import datetime, timezone
 
 
@@ -438,6 +439,9 @@ class MLModelConfig(ApiDto):
     :ivar str source: define name of model storage to use 'wizata'.
     :ivar str|list target_feat: target feature name(s) to remove from df in 'predict' mode.
     :ivar str train_script: name of function referencing the script to train the model.
+    :ivar ScriptType train_script_type: the trainer's type (JSON ``train_script_type``): ``raw``
+        for a trainer kept in the attached git repository; when absent it is inferred, built-in
+        (``wizata.*``) or uploaded with the SDK (DEV-6125).
     :ivar float train_test_split_pct: percentage repartition to split the data for training and scoring.
     :ivar str train_test_split_type: type of splitting desired to split the train dataframe.
     :ivar bool output_append: true - default - append output to input dataframe. false to retrieve only 'predict' column(s).
@@ -464,11 +468,13 @@ class MLModelConfig(ApiDto):
                  by_twin: bool = False,
                  by_property: bool = False,
                  property_name: str = None,
-                 source: str = "wizata"
+                 source: str = "wizata",
+                 train_script_type=None
                  ):
 
         # training
         self.train_script = train_script
+        self.train_script_type = ScriptType(train_script_type) if train_script_type is not None else None
         self.train_test_split_pct = train_test_split_pct
         self.train_test_split_type = train_test_split_type
         self.function = function
@@ -535,6 +541,8 @@ class MLModelConfig(ApiDto):
         # training info
         if "train_script" in obj.keys() and obj["train_script"] is not None:
             self.train_script = obj["train_script"]
+        if obj.get("train_script_type") is not None:
+            self.train_script_type = ScriptType(obj["train_script_type"])
         if "train_test_split_pct" in obj.keys() and obj["train_test_split_pct"] is not None:
             self.train_test_split_pct = float(obj["train_test_split_pct"])
         if "train_test_split_type" in obj.keys() and obj["train_test_split_type"] is not None:
@@ -612,6 +620,8 @@ class MLModelConfig(ApiDto):
         # training info
         if self.train_script is not None:
             obj["train_script"] = str(self.train_script)
+        if getattr(self, "train_script_type", None) is not None:
+            obj["train_script_type"] = self.train_script_type.value
         if self.train_test_split_pct is not None:
             obj["train_test_split_pct"] = float(self.train_test_split_pct)
         if self.train_test_split_type is not None:

wizata_dsapi-2.2.0.dev28/wizata_dsapi/version.py

@@ -0,0 +1 @@
+__version__ = "2.2.0.dev28"

{wizata_dsapi-2.2.0.dev27 → wizata_dsapi-2.2.0.dev28}/wizata_dsapi.egg-info/PKG-INFO

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

wizata_dsapi-2.2.0.dev27/wizata_dsapi/version.py

@@ -1 +0,0 @@
-__version__ = "2.2.0.dev27"