sinapsis-data-analysis 0.1.10__py3-none-any.whl → 0.1.11__py3-none-any.whl

sinapsis_data_analysis/templates/ml_base_inference.py

@@ -1,4 +1,5 @@
 # -*- coding: utf-8 -*-
+import os
 from abc import abstractmethod
 from typing import Any
 
@@ -6,6 +7,7 @@ import numpy as np
 from sinapsis_core.data_containers.data_packet import DataContainer
 from sinapsis_core.template_base.base_models import TemplateAttributes
 from sinapsis_core.template_base.template import Template
+from sinapsis_core.utils.env_var_keys import SINAPSIS_CACHE_DIR
 
 
 class MLBaseInference(Template):
@@ -24,11 +26,12 @@ class MLBaseInference(Template):
         """
 
         model_path: str
+        root_dir : str = SINAPSIS_CACHE_DIR
         generic_field_key: str
 
     def __init__(self, attributes: TemplateAttributes) -> None:
         super().__init__(attributes)
-        self.model = self.load_model(self.attributes.model_path)
+
 
     def get_data(self, container: DataContainer) -> Any:
         """Get the data from the data container
@@ -111,7 +114,8 @@ class MLBaseInference(Template):
         if not self.data_is_valid(data):
             self.logger.warning("Invalid or missing data")
             return container
-
+        full_path = os.path.join(self.attributes.root_dir, self.attributes.model_path)
+        self.model = self.load_model(full_path)
         data = self.preprocess_data(data)
         predictions = self.predict(data)
 

sinapsis_data_analysis/templates/ml_base_training.py

@@ -34,7 +34,7 @@ class MLBaseAttributes(TemplateAttributes):
         model_save_path (str): Path where the trained model will be saved.
     """
 
-    generic_field_key: str
+    generic_field_key: str | None = None
     root_dir : str = WORKING_DIR
     model_save_path: str
 
@@ -220,13 +220,13 @@ class MLBaseTraining(BaseDynamicWrapperTemplate):
         full_path = os.path.join(self.attributes.root_dir, self.attributes.model_save_path)
         try:
             os.makedirs(os.path.dirname(full_path), exist_ok=True)
-            self._save_model_implementation()
+            self._save_model_implementation(full_path)
             self.logger.info(f"Model saved at {self.attributes.model_save_path}")
         except (MemoryError, TypeError) as e:
             self.logger.error(f"Error saving model: {e}")
 
     @abstractmethod
-    def _save_model_implementation(self) -> None:
+    def _save_model_implementation(self, full_path: str) -> None:
         """Save the trained model using an implementation-specific method.
 
         This abstract method should be implemented by subclasses to define

sinapsis_data_analysis/templates/sklearn_train.py

@@ -53,12 +53,12 @@ class SKLearnLinearModelsTrain(MLBaseTraining):
         category="SKLearn", tags=[Tags.DATA_ANALYSIS, Tags.LINEAR_REGRESSION, Tags.MODELS, Tags.SKLEARN, Tags.TRAINING]
     )
 
-    def _save_model_implementation(self) -> None:
+    def _save_model_implementation(self, full_path:str) -> None:
         """
         Implements the abstract method from the base class to
         save the model to the path specified in attributes.
         """
-        joblib.dump(self.trained_model, self.attributes.model_save_path)
+        joblib.dump(self.trained_model, full_path)
 
 
 class SKLearnNeighborsModelsTrain(SKLearnLinearModelsTrain):

{sinapsis_data_analysis-0.1.10.dist-info → sinapsis_data_analysis-0.1.11.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: sinapsis-data-analysis
-Version: 0.1.10
+Version: 0.1.11
 Summary: Templates to work with models for classification, regression and clustering with xgboost and sklearn.
 Author-email: SinapsisAI <dev@sinapsis.tech>
 Project-URL: Homepage, https://sinapsis.tech

{sinapsis_data_analysis-0.1.10.dist-info → sinapsis_data_analysis-0.1.11.dist-info}/RECORD

@@ -4,15 +4,15 @@ sinapsis_data_analysis/helpers/excluded_models.py,sha256=gVrTRBdktM3eoBxd7kX22xH
 sinapsis_data_analysis/helpers/model_metrics.py,sha256=mR-ZLD8PrGgOh1PrYG8TOjFhJz6l2EDRtn6jtMMyI7A,729
 sinapsis_data_analysis/helpers/tags.py,sha256=ICJNmbNAtwn5BfMvGHh6eLuq5Ro3hCIbwd-Y63qOmC0,372
 sinapsis_data_analysis/templates/__init__.py,sha256=z1-E6V4f6MqoGY0_cPwSurDpXDdMOAEOvSS4ogrl1DM,840
-sinapsis_data_analysis/templates/ml_base_inference.py,sha256=CMAebWYKAyNWWdJFEc4zAjFH5z5RlEbsunvYhlunYno,3467
-sinapsis_data_analysis/templates/ml_base_training.py,sha256=3-YfNie6kyWHmpn9Dn2okfpOoGQsjY1iH-a3hwxt5aw,9325
+sinapsis_data_analysis/templates/ml_base_inference.py,sha256=3QfMH8w4yNCnEFZeJOOO3HZAC75UjJL2P6R4aN06f9g,3655
+sinapsis_data_analysis/templates/ml_base_training.py,sha256=m2yDuR9C7lfN1jT31fTQsn3c4fN0Gtx5jd8MtaemUqw,9364
 sinapsis_data_analysis/templates/sklearn_inference.py,sha256=rf-HsWYE4_g9KM_BZbP3f2jKyKz_MLj9-eZ1CAGHqp8,931
 sinapsis_data_analysis/templates/sklearn_manifold.py,sha256=whYTvwDnVacSh4N2XBXVgn8O1dG-KnQsx6htm5DFEbQ,5519
-sinapsis_data_analysis/templates/sklearn_train.py,sha256=WIF7DIslVIRHCmDAXfVUZkiVtLRPRbg6Snq0Di7Ln3w,6757
+sinapsis_data_analysis/templates/sklearn_train.py,sha256=uReIOKQTHYbfsdNy1efVcguwsMTkhDXTbBpSeVDlP-E,6750
 sinapsis_data_analysis/templates/xgboost_inference.py,sha256=rFk44jHFCr-EuqDg74TInjlHDCstvA0V422gQmgbd0I,1028
 sinapsis_data_analysis/templates/xgboost_train.py,sha256=VQSztloGAW4Xcx8MZCoRKdqa5jn4NL857jHTeMRuMwA,2513
-sinapsis_data_analysis-0.1.10.dist-info/licenses/LICENSE,sha256=hIahDEOTzuHCU5J2nd07LWwkLW7Hko4UFO__ffsvB-8,34523
-sinapsis_data_analysis-0.1.10.dist-info/METADATA,sha256=4LiJ3H5ka3Ocimbxt4F52ohKG2yjgbMVoyOoSzSlcbM,6297
-sinapsis_data_analysis-0.1.10.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-sinapsis_data_analysis-0.1.10.dist-info/top_level.txt,sha256=Mc5OyqBINgXFLrAyVBmjg25MQd6Lbg7z-rwotzEeygQ,23
-sinapsis_data_analysis-0.1.10.dist-info/RECORD,,
+sinapsis_data_analysis-0.1.11.dist-info/licenses/LICENSE,sha256=hIahDEOTzuHCU5J2nd07LWwkLW7Hko4UFO__ffsvB-8,34523
+sinapsis_data_analysis-0.1.11.dist-info/METADATA,sha256=OmUDyt8OsaKGfmEmytn_7bgi6pQLpGwVudOrqyauu_I,6297
+sinapsis_data_analysis-0.1.11.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
+sinapsis_data_analysis-0.1.11.dist-info/top_level.txt,sha256=Mc5OyqBINgXFLrAyVBmjg25MQd6Lbg7z-rwotzEeygQ,23
+sinapsis_data_analysis-0.1.11.dist-info/RECORD,,

{sinapsis_data_analysis-0.1.10.dist-info → sinapsis_data_analysis-0.1.11.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (80.9.0)
+Generator: setuptools (80.10.2)
 Root-Is-Purelib: true
 Tag: py3-none-any