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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dragon-ml-toolbox
-Version: 13.2.1
+Version: 13.3.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.1 → dragon_ml_toolbox-13.3.1/dragon_ml_toolbox.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dragon-ml-toolbox
-Version: 13.2.1
+Version: 13.3.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.1 → dragon_ml_toolbox-13.3.1}/ml_tools/_schema.py

@@ -83,3 +83,14 @@ class FeatureSchema(NamedTuple):
         self.save_all_features(directory=directory, verbose=True)
         self.save_continuous_features(directory=directory, verbose=True)
         self.save_categorical_features(directory=directory, verbose=True)
+        
+    def __repr__(self) -> str:
+        """Returns a concise representation of the schema's contents."""
+        total = len(self.feature_names)
+        cont = len(self.continuous_feature_names)
+        cat = len(self.categorical_feature_names)
+        index_map = self.categorical_index_map is not None
+        cat_map = self.categorical_mappings is not None
+        return (
+            f"<FeatureSchema(total={total}, continuous={cont}, categorical={cat}, index_map={index_map}, categorical_map={cat_map})>"
+        )

{dragon_ml_toolbox-13.2.1 → dragon_ml_toolbox-13.3.1}/ml_tools/data_exploration.py

@@ -1262,7 +1262,7 @@ def finalize_feature_schema(
     categorical_feature_names_list: List[str] = []
     categorical_index_map_dict: Dict[int, int] = {}
 
-    _LOGGER.info("Finalizing feature schema...")
+    # _LOGGER.info("Finalizing feature schema...")
 
     if categorical_mappings:
         # --- Categorical features are present ---

{dragon_ml_toolbox-13.2.1 → dragon_ml_toolbox-13.3.1}/ml_tools/serde.py

@@ -18,6 +18,10 @@ __all__ = [
 ]
 
 
+# Base types that have a generic `type()` log.
+_SIMPLE_TYPES = (list, dict, tuple, set, str, int, float, bool)
+
+
 def serialize_object_filename(obj: Any, save_dir: Union[str,Path], filename: str, verbose: bool=True, raise_on_error: bool=False) -> None:
     """
     Serializes a Python object using joblib; suitable for Python built-ins, numpy, and pandas.
@@ -27,6 +31,10 @@ def serialize_object_filename(obj: Any, save_dir: Union[str,Path], filename: str
         save_dir (str | Path) : Directory path where the serialized object will be saved.
         filename (str) : Name for the output file, extension will be appended if needed.
     """
+    if obj is None:
+        _LOGGER.warning(f"Attempted to serialize a None object. Skipping save for '{filename}'.")
+        return
+    
     try:
         save_path = make_fullpath(save_dir, make=True, enforce="directory")
         sanitized_name = sanitize_filename(filename)
@@ -56,6 +64,10 @@ def serialize_object(obj: Any, file_path: Path, verbose: bool = True, raise_on_e
                            '.joblib' extension will be appended if missing.
         raise_on_error (bool) : If True, raises exceptions on failure.
     """
+    if obj is None:
+        _LOGGER.warning(f"Attempted to serialize a None object. Skipping save for '{file_path}'.")
+        return
+    
     try:
         # Ensure the extension is correct
         file_path = file_path.with_suffix('.joblib')
@@ -73,7 +85,11 @@ def serialize_object(obj: Any, file_path: Path, verbose: bool = True, raise_on_e
         return None
     else:
         if verbose:
-            _LOGGER.info(f"Object of type '{type(obj)}' saved to '{file_path}'")
+            if type(obj) in _SIMPLE_TYPES:
+                _LOGGER.info(f"Object of type '{type(obj)}' saved to '{file_path}'")
+            else:
+                _LOGGER.info(f"Object '{obj}' saved to '{file_path}'")
+
         return None
 
 
@@ -123,9 +139,13 @@ def deserialize_object(
                 raise TypeError()
         
         if verbose:
-            _LOGGER.info(f"Loaded object of type '{type(obj)}' from '{true_filepath}'.")
+            # log special objects
+            if type(obj) in _SIMPLE_TYPES:
+                _LOGGER.info(f"Loaded object of type '{type(obj)}' from '{true_filepath}'.")
+            else:
+                _LOGGER.info(f"Loaded object '{obj}' from '{true_filepath}'.")
         
-        return obj
+        return obj # type: ignore
 
 
 def serialize_schema(schema: FeatureSchema, file_path: Path):

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

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