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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: wizata_dsapi
-Version: 2.0.0.dev19
+Version: 2.0.0.dev21
 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.dev21}/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.dev21}/wizata_dsapi/pipeline.py

@@ -351,7 +351,9 @@ class PipelineStep(ApiDto):
                  step_type: StepType = None,
                  config=None,
                  inputs=None,
-                 outputs=None):
+                 outputs=None,
+                 name: str = None,
+                 description: str = None):
 
         if outputs is None:
             outputs = []
@@ -361,6 +363,8 @@ class PipelineStep(ApiDto):
             step_id = uuid.uuid4()
         self.step_id = step_id
         self.step_type = step_type
+        self.name = name
+        self.description = description
         self.config = config
         self.inputs = inputs
         self.outputs = outputs
@@ -368,6 +372,10 @@ class PipelineStep(ApiDto):
     def from_json(self, obj):
         if "id" in obj.keys():
             self.step_id = uuid.UUID(obj["id"])
+        if "name" in obj:
+            self.name = obj["name"]
+        if "description" in obj:
+            self.description = obj["description"]
 
         if "type" in obj.keys():
             self.step_type = StepType(obj["type"])
@@ -408,6 +416,10 @@ class PipelineStep(ApiDto):
         }
         if self.step_type is not None:
             obj["type"] = str(self.step_type.value)
+        if self.name is not None:
+            obj["name"] = self.name
+        if self.description is not None:
+            obj["description"] = self.description
         if self.config is not None:
             if self.step_type == StepType.SCRIPT:
                 obj["config"] = self.config.to_json()

{wizata_dsapi-2.0.0.dev19 → wizata_dsapi-2.0.0.dev21}/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.dev21/wizata_dsapi/version.py

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

{wizata_dsapi-2.0.0.dev19 → wizata_dsapi-2.0.0.dev21/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.dev21
 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"