wizata-dsapi 2.0.0.dev22__tar.gz → 2.0.0.dev23__tar.gz

{wizata_dsapi-2.0.0.dev22/wizata_dsapi.egg-info → wizata_dsapi-2.0.0.dev23}/PKG-INFO

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

{wizata_dsapi-2.0.0.dev22 → wizata_dsapi-2.0.0.dev23}/wizata_dsapi/mlmodel.py

@@ -472,6 +472,7 @@ class MLModelConfig(ApiDto):
         self.train_test_split_pct = train_test_split_pct
         self.train_test_split_type = train_test_split_type
         self.function = function
+        self.properties = None
         self.properties_mapping = properties_mapping
 
         # features management
@@ -540,6 +541,8 @@ class MLModelConfig(ApiDto):
             self.train_test_split_type = obj["train_test_split_type"]
         if "function" in obj.keys() and obj["function"] is not None:
             self.function = obj["function"]
+        if "properties" in obj and obj["properties"] is not None:
+            self.properties = obj["properties"]
         if "properties_mapping" in obj:
             self.properties_mapping = obj["properties_mapping"]
 
@@ -617,6 +620,8 @@ class MLModelConfig(ApiDto):
             obj["features"] = self.features
         if self.features_from_file is not None:
             obj["features_from_file"] = self.features_from_file
+        if self.properties is not None and isinstance(self.properties, dict):
+            obj["properties"] = self.properties
         if self.properties_mapping is not None and isinstance(self.properties_mapping, dict):
             obj["properties_mapping"] = self.properties_mapping
 

{wizata_dsapi-2.0.0.dev22 → wizata_dsapi-2.0.0.dev23}/wizata_dsapi/script.py

@@ -187,7 +187,8 @@ class Script(ApiDto):
         return "json"
 
     def __init__(self, script_id=None, description=None, function=None,
-                 script_type=None, version=None, library=None, category=None):
+                 script_type=None, version=None, library=None, category=None,
+                 properties=None):
 
         # Id
         if script_id is None:
@@ -202,6 +203,7 @@ class Script(ApiDto):
         self.library = library
         self.module = None
         self.category = category
+        self.properties = properties if properties is not None else {}
 
         # Source code property (not serialized)
         self.source = None
@@ -271,6 +273,8 @@ class Script(ApiDto):
             obj["source"] = self.source
         if self.category is not None:
             obj["category"] = self.category.value
+        if self.properties:
+            obj["properties"] = self.properties
         return obj
 
     def to_json(self, target: str = None):
@@ -304,6 +308,8 @@ class Script(ApiDto):
             self.source = obj["source"]
         if "category" in obj and obj["category"] is not None:
             self.category = ScriptCategory(obj["category"])
+        if "properties" in obj and obj["properties"] is not None:
+            self.properties = obj["properties"]
 
     def copy(self, myfunction):
         """

wizata_dsapi-2.0.0.dev23/wizata_dsapi/version.py

@@ -0,0 +1 @@
+__version__ = "2.0.0.dev23"

{wizata_dsapi-2.0.0.dev22 → wizata_dsapi-2.0.0.dev23/wizata_dsapi.egg-info}/PKG-INFO

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

wizata_dsapi-2.0.0.dev22/wizata_dsapi/version.py

@@ -1 +0,0 @@
-__version__ = "2.0.0.dev22"