dragon-ml-toolbox 13.2.0__tar.gz → 13.2.1__tar.gz

{dragon_ml_toolbox-13.2.0/dragon_ml_toolbox.egg-info → dragon_ml_toolbox-13.2.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dragon-ml-toolbox
-Version: 13.2.0
+Version: 13.2.1
 Summary: A collection of tools for data science and machine learning projects.
 Author-email: "Karl L. Loza Vidaurre" <luigiloza@gmail.com>
 License-Expression: MIT

{dragon_ml_toolbox-13.2.0 → dragon_ml_toolbox-13.2.1/dragon_ml_toolbox.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dragon-ml-toolbox
-Version: 13.2.0
+Version: 13.2.1
 Summary: A collection of tools for data science and machine learning projects.
 Author-email: "Karl L. Loza Vidaurre" <luigiloza@gmail.com>
 License-Expression: MIT

{dragon_ml_toolbox-13.2.0 → dragon_ml_toolbox-13.2.1}/ml_tools/_schema.py

@@ -3,6 +3,7 @@ from pathlib import Path
 
 from .custom_logger import save_list_strings
 from .keys import DatasetKeys
+from ._logger import _LOGGER
 
 
 class FeatureSchema(NamedTuple):
@@ -25,6 +26,12 @@ class FeatureSchema(NamedTuple):
 
     def _save_helper(self, artifact: Tuple[str, ...], directory: Union[str,Path], filename: str, verbose: bool):
         to_save = list(artifact)
+        
+        # empty check
+        if not to_save:
+            _LOGGER.warning(f"Skipping save for '{filename}': The feature list is empty.")
+            return
+        
         save_list_strings(list_strings=to_save,
                           directory=directory,
                           filename=filename,
@@ -68,3 +75,11 @@ class FeatureSchema(NamedTuple):
                           directory=directory,
                           filename=DatasetKeys.CATEGORICAL_NAMES,
                           verbose=verbose)
+        
+    def save_artifacts(self, directory: Union[str,Path]):
+        """
+        Saves feature names, categorical feature names, continuous feature names to separate text files.
+        """
+        self.save_all_features(directory=directory, verbose=True)
+        self.save_continuous_features(directory=directory, verbose=True)
+        self.save_categorical_features(directory=directory, verbose=True)

{dragon_ml_toolbox-13.2.0 → dragon_ml_toolbox-13.2.1}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "dragon-ml-toolbox"
-version = "13.2.0"
+version = "13.2.1"
 description = "A collection of tools for data science and machine learning projects."
 authors = [
     { name = "Karl L. Loza Vidaurre", email = "luigiloza@gmail.com" }