wizata-dsapi 1.3.18__tar.gz → 1.3.19__tar.gz

{wizata_dsapi-1.3.18/wizata_dsapi.egg-info → wizata_dsapi-1.3.19}/PKG-INFO

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

{wizata_dsapi-1.3.18 → wizata_dsapi-1.3.19}/wizata_dsapi/context.py

@@ -1,7 +1,7 @@
 import uuid
 import pandas
 
-from .mlmodel import MLModel
+from .mlmodel import ModelInfo
 from .plot import Plot
 from .ilogger import ILogger
 from .api_interface import ApiInterface
@@ -198,16 +198,16 @@ class Context(ILogger):
         :param scaler: Scaler to be stored if necessary.
         :return: ML Model object prepared.
         """
-        ml_model = MLModel()
-        ml_model.trained_model = trained_model
-        ml_model.scaler = scaler
-        ml_model.input_columns = input_columns
-        ml_model.output_columns = output_columns
-        ml_model.has_anomalies = has_anomalies
-        self.__new_model = ml_model
-        return ml_model
+        model_info = ModelInfo(key="undefined")
+        model_info.trained_model = trained_model
+        model_info.scaler = scaler
+        model_info.input_columns = input_columns
+        model_info.output_columns = output_columns
+        model_info.has_anomalies = has_anomalies
+        self.__new_model = model_info
+        return model_info
 
-    def get_model(self) -> MLModel:
+    def get_model(self) -> ModelInfo:
         """
         get model to be added to the context.
         """
@@ -280,10 +280,10 @@ class Context(ILogger):
             raise ValueError(f'this context is not running within an execution (execution_id is None).')
         if model_key not in self.models:
             raise RuntimeError(f'the model {model_key} is not present within the context.')
-        model: MLModel = self.models[model_key]
+        model: ModelInfo = self.models[model_key]
         evaluation = Evaluation(
             execution_id=self.execution_id,
-            model_id=model.model_id,
+            model_id=model.identifier(),
             properties=properties,
             metrics=metrics
         )

wizata_dsapi-1.3.19/wizata_dsapi/version.py

@@ -0,0 +1 @@
+__version__ = "1.3.19"

{wizata_dsapi-1.3.18 → wizata_dsapi-1.3.19/wizata_dsapi.egg-info}/PKG-INFO

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

wizata_dsapi-1.3.18/wizata_dsapi/version.py

@@ -1 +0,0 @@
-__version__ = "1.3.18"