wizata-dsapi 2.0.0.dev19__tar.gz → 2.0.0.dev20__tar.gz

{wizata_dsapi-2.0.0.dev19/wizata_dsapi.egg-info → wizata_dsapi-2.0.0.dev20}/PKG-INFO

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

{wizata_dsapi-2.0.0.dev19 → wizata_dsapi-2.0.0.dev20}/wizata_dsapi/__init__.py

@@ -9,7 +9,7 @@ from .execution import Execution, ExecutionStatus, ExecutionStepLog, AbortedExce
 from .execution_log import ExecutionLog
 from .experiment import Experiment
 from .ds_dataframe import DSDataFrame
-from .script import Script, ScriptConfig, ScriptType, ScriptLibrary, sanitize_blob_name, discover_from_wheel
+from .script import Script, ScriptConfig, ScriptType, ScriptCategory, ScriptLibrary, sanitize_blob_name, discover_from_wheel
 from .template import Template, TemplateProperty
 from .solution_component import SolutionComponent, SolutionType
 from .business_label import BusinessLabel

{wizata_dsapi-2.0.0.dev19 → wizata_dsapi-2.0.0.dev20}/wizata_dsapi/script.py

@@ -25,6 +25,18 @@ class ScriptType(Enum):
     RAW = 'raw'
 
 
+class ScriptCategory(Enum):
+    """
+    Defines the category of a script, determining which pipeline step type it creates.
+        - "function" data transformation scripts (creates a SCRIPT step)
+        - "model" ML model training scripts (creates a MODEL step)
+        - "plot" visualization scripts (creates a PLOT step)
+    """
+    FUNCTION = 'function'
+    MODEL = 'model'
+    PLOT = 'plot'
+
+
 def sanitize_blob_name(name: str) -> str:
     """
     Sanitize a script name for use as a blob path segment.
@@ -175,7 +187,7 @@ class Script(ApiDto):
         return "json"
 
     def __init__(self, script_id=None, description=None, function=None,
-                 script_type=None, version=None, library=None):
+                 script_type=None, version=None, library=None, category=None):
 
         # Id
         if script_id is None:
@@ -189,6 +201,7 @@ class Script(ApiDto):
         self.version = version
         self.library = library
         self.module = None
+        self.category = category
 
         # Source code property (not serialized)
         self.source = None
@@ -256,6 +269,8 @@ class Script(ApiDto):
             obj["module"] = self.module
         if self.source is not None:
             obj["source"] = self.source
+        if self.category is not None:
+            obj["category"] = self.category.value
         return obj
 
     def to_json(self, target: str = None):
@@ -287,6 +302,8 @@ class Script(ApiDto):
             self.module = obj["module"]
         if "source" in obj and obj["source"] is not None:
             self.source = obj["source"]
+        if "category" in obj and obj["category"] is not None:
+            self.category = ScriptCategory(obj["category"])
 
     def copy(self, myfunction):
         """

wizata_dsapi-2.0.0.dev20/wizata_dsapi/version.py

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

{wizata_dsapi-2.0.0.dev19 → wizata_dsapi-2.0.0.dev20/wizata_dsapi.egg-info}/PKG-INFO

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

wizata_dsapi-2.0.0.dev19/wizata_dsapi/version.py

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