dragon-ml-toolbox 14.3.0__py3-none-any.whl → 14.3.1__py3-none-any.whl

{dragon_ml_toolbox-14.3.0.dist-info → dragon_ml_toolbox-14.3.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dragon-ml-toolbox
-Version: 14.3.0
+Version: 14.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-14.3.0.dist-info → dragon_ml_toolbox-14.3.1.dist-info}/RECORD

@@ -1,5 +1,5 @@
-dragon_ml_toolbox-14.3.0.dist-info/licenses/LICENSE,sha256=L35WDmmLZNTlJvxF6Vy7Uy4SYNi6rCfWUqlTHpoRMoU,1081
-dragon_ml_toolbox-14.3.0.dist-info/licenses/LICENSE-THIRD-PARTY.md,sha256=gkOdNDbKYpIJezwSo2CEnISkLeYfYHv9t8b5K2-P69A,2687
+dragon_ml_toolbox-14.3.1.dist-info/licenses/LICENSE,sha256=L35WDmmLZNTlJvxF6Vy7Uy4SYNi6rCfWUqlTHpoRMoU,1081
+dragon_ml_toolbox-14.3.1.dist-info/licenses/LICENSE-THIRD-PARTY.md,sha256=gkOdNDbKYpIJezwSo2CEnISkLeYfYHv9t8b5K2-P69A,2687
 ml_tools/ETL_cleaning.py,sha256=2VBRllV8F-ZiPylPp8Az2gwn5ztgazN0BH5OKnRUhV0,20402
 ml_tools/ETL_engineering.py,sha256=KfYqgsxupAx6e_TxwO1LZXeu5mFkIhVXJrNjP3CzIZc,54927
 ml_tools/GUI_tools.py,sha256=Va6ig-dHULPVRwQYYtH3fvY5XPIoqRcJpRW8oXC55Hw,45413
@@ -15,7 +15,7 @@ ml_tools/ML_optimization.py,sha256=P0zkhKAwTpkorIBtR0AOIDcyexo5ngmvFUzo3DfNO-E,2
 ml_tools/ML_scaler.py,sha256=tw6onj9o8_kk3FQYb930HUzvv1zsFZe2YZJdF3LtHkU,7538
 ml_tools/ML_trainer.py,sha256=ZWI4MbUcLeBxyfoUTL96l5tjHHMp9I64h4SdXnjYmBE,49795
 ml_tools/ML_utilities.py,sha256=z6LbpbZwhn8F__fWlKi-g-cAJQXSxwg1NHfC5FBoAyc,21139
-ml_tools/ML_vision_datasetmaster.py,sha256=feFNUBjybzVJJrdyqToQ_mLV1uDJXHkNL0tmn_zofSY,56034
+ml_tools/ML_vision_datasetmaster.py,sha256=2S7stCgGQX39Y38gfMQccwAx_QTKEmmRIqh4XJ2K_YE,58041
 ml_tools/ML_vision_evaluation.py,sha256=t12R7i1RkOCt9zu1_lxSBr8OH6A6Get0k8ftDLctn6I,10486
 ml_tools/ML_vision_inference.py,sha256=He3KV3VJAm8PwO-fOq4b9VO8UXFr-GmpuCnoHXf4VZI,20588
 ml_tools/ML_vision_models.py,sha256=G3S4jB9AE9wMpU9ZygOgOx9q1K6t6LAXBYcJ-U2XQ1M,25600
@@ -42,7 +42,7 @@ ml_tools/optimization_tools.py,sha256=TYFQ2nSnp7xxs-VyoZISWgnGJghFbsWasHjruegyJR
 ml_tools/path_manager.py,sha256=CyDU16pOKmC82jPubqJPT6EBt-u-3rGVbxyPIZCvDDY,18432
 ml_tools/serde.py,sha256=c8uDYjYry_VrLvoG4ixqDj5pij88lVn6Tu4NHcPkwDU,6943
 ml_tools/utilities.py,sha256=aWqvYzmxlD74PD5Yqu1VuTekDJeYLQrmPIU_VeVyRp0,22526
-dragon_ml_toolbox-14.3.0.dist-info/METADATA,sha256=TeVrfmCt4AVSweSN4Ai0yyZCJMQtSD1MHsUoEQHXLg4,6475
-dragon_ml_toolbox-14.3.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-dragon_ml_toolbox-14.3.0.dist-info/top_level.txt,sha256=wm-oxax3ciyez6VoO4zsFd-gSok2VipYXnbg3TH9PtU,9
-dragon_ml_toolbox-14.3.0.dist-info/RECORD,,
+dragon_ml_toolbox-14.3.1.dist-info/METADATA,sha256=dNN-vygEF2WikswJ-6XCShUDf3rD2-XRNtz0vvRg2-4,6475
+dragon_ml_toolbox-14.3.1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+dragon_ml_toolbox-14.3.1.dist-info/top_level.txt,sha256=wm-oxax3ciyez6VoO4zsFd-gSok2VipYXnbg3TH9PtU,9
+dragon_ml_toolbox-14.3.1.dist-info/RECORD,,

ml_tools/ML_vision_datasetmaster.py

@@ -10,6 +10,7 @@ import torchvision.transforms.functional as TF
 from pathlib import Path
 import random
 import json
+import inspect
 
 from .ML_datasetmaster import _BaseMaker
 from .path_manager import make_fullpath
@@ -411,16 +412,58 @@ class VisionDatasetMaker(_BaseMaker):
         # validate path
         file_path = make_fullpath(filepath, make=True, enforce="file")
 
-        # 1. Handle pre_transforms
+        # Handle pre_transforms
         for t in components[VisionTransformRecipeKeys.PRE_TRANSFORMS]:
             t_name = t.__class__.__name__
+            t_class = t.__class__
+            kwargs = {}
+            
+            # 1. Check custom registry first
             if t_name in TRANSFORM_REGISTRY:
-                recipe[VisionTransformRecipeKeys.PIPELINE].append({
-                    VisionTransformRecipeKeys.NAME: t_name,
-                    VisionTransformRecipeKeys.KWARGS: getattr(t, VisionTransformRecipeKeys.KWARGS, {})
-                })
+                _LOGGER.debug(f"Found '{t_name}' in TRANSFORM_REGISTRY.")
+                kwargs = getattr(t, VisionTransformRecipeKeys.KWARGS, {})
+
+            # 2. Else, try to introspect for standard torchvision transforms
             else:
-                _LOGGER.warning(f"Skipping unknown pre_transform '{t_name}' in recipe. Not in TRANSFORM_REGISTRY.")
+                _LOGGER.debug(f"'{t_name}' not in registry. Attempting introspection...")
+                try:
+                    # Get the __init__ signature of the transform's class
+                    sig = inspect.signature(t_class.__init__)
+                    
+                    # Iterate over its __init__ parameters (e.g., 'num_output_channels')
+                    for param in sig.parameters.values():
+                        if param.name == 'self':
+                            continue
+                        
+                        # Check if the *instance* 't' has that parameter as an attribute
+                        attr_name_public = param.name
+                        attr_name_private = '_' + param.name
+                        
+                        attr_to_get = ""
+                        
+                        if hasattr(t, attr_name_public):
+                            attr_to_get = attr_name_public
+                        elif hasattr(t, attr_name_private):
+                            attr_to_get = attr_name_private
+                        else:
+                            # Parameter in __init__ has no matching attribute
+                            continue 
+                        
+                        # Store the value under the __init__ parameter's name
+                        kwargs[param.name] = getattr(t, attr_to_get)
+                            
+                    _LOGGER.debug(f"Introspection for '{t_name}' found kwargs: {kwargs}")
+
+                except (ValueError, TypeError):
+                    # Fails on some built-ins or C-implemented __init__
+                    _LOGGER.warning(f"Could not introspect parameters for '{t_name}'. If this transform has parameters, they will not be saved.")
+                    kwargs = {}
+
+            # 3. Add to pipeline
+            recipe[VisionTransformRecipeKeys.PIPELINE].append({
+                VisionTransformRecipeKeys.NAME: t_name,
+                VisionTransformRecipeKeys.KWARGS: kwargs
+            })
                 
         # 2. Add standard transforms
         recipe[VisionTransformRecipeKeys.PIPELINE].extend([