dragon-ml-toolbox 19.9.0__tar.gz → 19.11.0__tar.gz

{dragon_ml_toolbox-19.9.0 → dragon_ml_toolbox-19.11.0}/LICENSE-THIRD-PARTY.md

@@ -31,13 +31,9 @@ This project depends on the following third-party packages. Each is governed by
 - [colorlog](https://github.com/borntyping/python-colorlog/blob/main/LICENSE)
 - [evotorch](https://github.com/nnaisense/evotorch/blob/master/LICENSE)
 - [FreeSimpleGUI](https://github.com/spyoungtech/FreeSimpleGUI/blob/main/license.txt)
-- [nuitka](https://github.com/Nuitka/Nuitka/blob/main/LICENSE.txt)
 - [omegaconf](https://github.com/omry/omegaconf/blob/master/LICENSE)
-- [ordered-set](https://github.com/rspeer/ordered-set/blob/master/MIT-LICENSE)
-- [pyinstaller](https://github.com/pyinstaller/pyinstaller/blob/develop/COPYING.txt)
 - [pytorch_tabular](https://github.com/manujosephv/pytorch_tabular/blob/main/LICENSE)
 - [torchmetrics](https://github.com/Lightning-AI/torchmetrics/blob/master/LICENSE)
-- [zstandard](https://github.com/indygreg/python-zstandard/blob/main/LICENSE)
 - [captum](https://github.com/meta-pytorch/captum/blob/master/LICENSE)
 - [node](https://github.com/Qwicen/node/blob/master/LICENSE.md)
 - [pytorch-widedeep](https://github.com/jrzaurin/pytorch-widedeep?tab=readme-ov-file#license)

{dragon_ml_toolbox-19.9.0/dragon_ml_toolbox.egg-info → dragon_ml_toolbox-19.11.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dragon-ml-toolbox
-Version: 19.9.0
+Version: 19.11.0
 Summary: Complete pipelines and helper tools for data science and machine learning projects.
 Author-email: Karl Luigi Loza Vidaurre <luigiloza@gmail.com>
 License-Expression: MIT
@@ -90,12 +90,6 @@ Provides-Extra: gui-torch
 Requires-Dist: numpy<2.0; extra == "gui-torch"
 Requires-Dist: torch; extra == "gui-torch"
 Requires-Dist: FreeSimpleGUI>=5.2; extra == "gui-torch"
-Provides-Extra: pyinstaller
-Requires-Dist: pyinstaller; extra == "pyinstaller"
-Provides-Extra: nuitka
-Requires-Dist: nuitka; extra == "nuitka"
-Requires-Dist: zstandard; extra == "nuitka"
-Requires-Dist: ordered-set; extra == "nuitka"
 Dynamic: license-file
 
 # dragon-ml-toolbox
@@ -137,7 +131,7 @@ conda install -c conda-forge dragon-ml-toolbox
 
 ## Modular Installation
 
-This toolbox is designed as a collection of mutually exclusive environments due to conflicting core dependencies, except APP bundlers (PyInstaller/Nuitka).
+This toolbox is designed as a collection of mutually exclusive environments due to conflicting core dependencies.
 
 - Rule: Create a fresh virtual environment for each module to use.
 
@@ -330,18 +324,6 @@ schema
 
 ---
 
-### ⚒️ APP bundlers
-
-Dependencies required to compile applications, inference scripts, or GUIs into standalone executables (`.exe` or binary) for distribution. Choose your preferred backend:
-
-```Bash
-pip install "dragon-ml-toolbox[pyinstaller]"
-```
-
-```Bash
-pip install "dragon-ml-toolbox[nuitka]"
-```
-
 ## Usage
 
 After installation, import modules like this:

{dragon_ml_toolbox-19.9.0 → dragon_ml_toolbox-19.11.0}/README.md

@@ -37,7 +37,7 @@ conda install -c conda-forge dragon-ml-toolbox
 
 ## Modular Installation
 
-This toolbox is designed as a collection of mutually exclusive environments due to conflicting core dependencies, except APP bundlers (PyInstaller/Nuitka).
+This toolbox is designed as a collection of mutually exclusive environments due to conflicting core dependencies.
 
 - Rule: Create a fresh virtual environment for each module to use.
 
@@ -230,18 +230,6 @@ schema
 
 ---
 
-### ⚒️ APP bundlers
-
-Dependencies required to compile applications, inference scripts, or GUIs into standalone executables (`.exe` or binary) for distribution. Choose your preferred backend:
-
-```Bash
-pip install "dragon-ml-toolbox[pyinstaller]"
-```
-
-```Bash
-pip install "dragon-ml-toolbox[nuitka]"
-```
-
 ## Usage
 
 After installation, import modules like this:

{dragon_ml_toolbox-19.9.0 → dragon_ml_toolbox-19.11.0/dragon_ml_toolbox.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dragon-ml-toolbox
-Version: 19.9.0
+Version: 19.11.0
 Summary: Complete pipelines and helper tools for data science and machine learning projects.
 Author-email: Karl Luigi Loza Vidaurre <luigiloza@gmail.com>
 License-Expression: MIT
@@ -90,12 +90,6 @@ Provides-Extra: gui-torch
 Requires-Dist: numpy<2.0; extra == "gui-torch"
 Requires-Dist: torch; extra == "gui-torch"
 Requires-Dist: FreeSimpleGUI>=5.2; extra == "gui-torch"
-Provides-Extra: pyinstaller
-Requires-Dist: pyinstaller; extra == "pyinstaller"
-Provides-Extra: nuitka
-Requires-Dist: nuitka; extra == "nuitka"
-Requires-Dist: zstandard; extra == "nuitka"
-Requires-Dist: ordered-set; extra == "nuitka"
 Dynamic: license-file
 
 # dragon-ml-toolbox
@@ -137,7 +131,7 @@ conda install -c conda-forge dragon-ml-toolbox
 
 ## Modular Installation
 
-This toolbox is designed as a collection of mutually exclusive environments due to conflicting core dependencies, except APP bundlers (PyInstaller/Nuitka).
+This toolbox is designed as a collection of mutually exclusive environments due to conflicting core dependencies.
 
 - Rule: Create a fresh virtual environment for each module to use.
 
@@ -330,18 +324,6 @@ schema
 
 ---
 
-### ⚒️ APP bundlers
-
-Dependencies required to compile applications, inference scripts, or GUIs into standalone executables (`.exe` or binary) for distribution. Choose your preferred backend:
-
-```Bash
-pip install "dragon-ml-toolbox[pyinstaller]"
-```
-
-```Bash
-pip install "dragon-ml-toolbox[nuitka]"
-```
-
 ## Usage
 
 After installation, import modules like this:

{dragon_ml_toolbox-19.9.0 → dragon_ml_toolbox-19.11.0}/dragon_ml_toolbox.egg-info/requires.txt

@@ -80,14 +80,6 @@ shap
 colorlog
 pyarrow
 
-[nuitka]
-nuitka
-zstandard
-ordered-set
-
 [py-tab]
 omegaconf
 pytorch_tabular
-
-[pyinstaller]
-pyinstaller

{dragon_ml_toolbox-19.9.0 → dragon_ml_toolbox-19.11.0}/ml_tools/_core/_MICE_imputation.py

@@ -256,7 +256,7 @@ def run_mice_pipeline(df_path_or_dir: Union[str,Path], target_columns: list[str]
     if input_path.is_file():
         all_file_paths = [input_path]
     else:
-        all_file_paths = list(list_csv_paths(input_path).values())
+        all_file_paths = list(list_csv_paths(input_path, raise_on_empty=True).values())
     
     for df_path in all_file_paths:
         df: pd.DataFrame
@@ -461,7 +461,7 @@ class DragonMICE:
         if input_path.is_file():
             all_file_paths = [input_path]
         else:
-            all_file_paths = list(list_csv_paths(input_path).values())
+            all_file_paths = list(list_csv_paths(input_path, raise_on_empty=True).values())
         
         for df_path in all_file_paths:
             

{dragon_ml_toolbox-19.9.0 → dragon_ml_toolbox-19.11.0}/ml_tools/_core/_ML_configuration.py

@@ -6,7 +6,7 @@ import numpy as np
 from ._schema import FeatureSchema
 from ._script_info import _script_info
 from ._logger import get_logger
-from ._path_manager import sanitize_filename
+from ._path_manager import sanitize_filename, make_fullpath
 from ._keys import MLTaskKeys
 
 
@@ -701,11 +701,11 @@ class DragonParetoConfig(_BaseModelParams):
     def __init__(self,
                  save_directory: Union[str, Path],
                  target_objectives: Dict[str, Literal["min", "max"]],
-                 continuous_bounds_map: Union[Dict[str, Tuple[float, float]], Dict[str, List[float]]],
+                 continuous_bounds_map: Union[Dict[str, Tuple[float, float]], Dict[str, List[float]], str, Path],
                  columns_to_round: Optional[List[str]] = None,
-                 population_size: int = 400,
+                 population_size: int = 500,
                  generations: int = 1000,
-                 solutions_filename: str = "ParetoSolutions",
+                 solutions_filename: str = "NonDominatedSolutions",
                  float_precision: int = 4,
                  log_interval: int = 10,
                  plot_size: Tuple[int, int] = (10, 7),
@@ -718,7 +718,7 @@ class DragonParetoConfig(_BaseModelParams):
             save_directory (str | Path): Directory to save artifacts.
             target_objectives (Dict[str, "min"|"max"]): Dictionary mapping target names to optimization direction.
                 Example: {"price": "max", "error": "min"}
-            continuous_bounds_map (Dict): Bounds for continuous features {name: (min, max)}.
+            continuous_bounds_map (Dict): Bounds for continuous features {name: (min, max)}. Or a path/str to a directory containing the "optimization_bounds.json" file.
             columns_to_round (List[str] | None): List of continuous column names that should be rounded to the nearest integer.
             population_size (int): Size of the genetic population.
             generations (int): Number of generations to run.
@@ -729,7 +729,13 @@ class DragonParetoConfig(_BaseModelParams):
             plot_font_size (int): Font size for plot text.
             discretize_start_at_zero (bool): Categorical encoding start index. True=0, False=1.
         """
-        self.save_directory = save_directory
+        # Validate string or Path
+        valid_save_dir = make_fullpath(save_directory, make=True, enforce="directory")
+        
+        if isinstance(continuous_bounds_map, (str, Path)):
+            continuous_bounds_map = make_fullpath(continuous_bounds_map, make=False, enforce="directory")
+        
+        self.save_directory = valid_save_dir
         self.target_objectives = target_objectives
         self.continuous_bounds_map = continuous_bounds_map
         self.columns_to_round = columns_to_round
@@ -742,7 +748,6 @@ class DragonParetoConfig(_BaseModelParams):
         self.plot_font_size = plot_font_size
         self.discretize_start_at_zero = discretize_start_at_zero
 
-
 # ----------------------------
 # Metrics Configurations
 # ----------------------------

{dragon_ml_toolbox-19.9.0 → dragon_ml_toolbox-19.11.0}/ml_tools/_core/_ML_optimization.py

@@ -44,12 +44,10 @@ class DragonOptimizer:
     SNES and CEM algorithms do not accept bounds, the given bounds will be used as an initial starting point.
 
     Example:
-        >>> # 1. Get the final schema from data exploration
-        >>> schema = data_exploration.finalize_feature_schema(...)
-        >>> # 2. Define bounds for continuous features
+        >>> # 1. Define bounds for continuous features
         >>> cont_bounds = {'feature_A': (0, 100), 'feature_B': (-10, 10)}
         >>>
-        >>> # 3. Initialize the optimizer
+        >>> # 2. Initialize the optimizer
         >>> optimizer = DragonOptimizer(
         ...     inference_handler=my_handler,
         ...     schema=schema,
@@ -58,7 +56,7 @@ class DragonOptimizer:
         ...     task="max",
         ...     algorithm="Genetic",
         ... )
-        >>> # 4. Run the optimization
+        >>> # 3. Run the optimization
         >>> best_result = optimizer.run(
         ...     num_generations=100,
         ...     save_dir="/path/to/results",

{dragon_ml_toolbox-19.9.0 → dragon_ml_toolbox-19.11.0}/ml_tools/_core/_ML_optimization_pareto.py

@@ -7,7 +7,7 @@ import matplotlib.cm as cm
 from matplotlib.collections import LineCollection
 import seaborn as sns
 from pathlib import Path
-from typing import Literal, Union, Tuple, List, Optional, Dict
+from typing import Literal, Union, List, Optional, Dict
 from tqdm import tqdm
 import plotly.express as px
 import plotly.graph_objects as go
@@ -21,7 +21,7 @@ from ._SQL import DragonSQL
 from ._ML_inference import DragonInferenceHandler
 from ._ML_chaining_inference import DragonChainInference
 from ._ML_configuration import DragonParetoConfig
-from ._optimization_tools import create_optimization_bounds, plot_optimal_feature_distributions_from_dataframe
+from ._optimization_tools import create_optimization_bounds, plot_optimal_feature_distributions_from_dataframe, load_continuous_bounds_template
 from ._math_utilities import discretize_categorical_values
 from ._utilities import save_dataframe_filename
 from ._IO_tools import save_json
@@ -107,6 +107,10 @@ class DragonParetoOptimizer:
                 _LOGGER.error(f"Target '{name}' not found in model targets: {available_targets}")
                 raise ValueError()
             
+            if direction not in ["min" , "max"]:
+                _LOGGER.error(f"Invalid optimization direction '{direction}' for target '{name}'. Use 'min' or 'max'.")
+                raise ValueError()
+            
             # For standard handlers, we need indices to slice the output tensor.
             # For chain handlers, we just rely on name matching, but we track index for consistency.
             idx = available_targets.index(name)
@@ -117,10 +121,20 @@ class DragonParetoOptimizer:
         _LOGGER.info(f"Pareto Optimization setup for: {self.ordered_target_names}")
 
         # --- 2. Bounds Setup ---
+        # check type
+        raw_bounds_map = config.continuous_bounds_map
+        if isinstance(raw_bounds_map, (str, Path)):
+            continuous_bounds = load_continuous_bounds_template(raw_bounds_map)
+        elif isinstance(raw_bounds_map, dict):
+            continuous_bounds = raw_bounds_map
+        else:
+            _LOGGER.error(f"Invalid type for 'continuous_bounds_map' in config. Expected dict or Path. Got {type(raw_bounds_map)}.")
+            raise ValueError()
+        
         # Uses the external tool which reads the schema to set correct bounds for both continuous and categorical
         bounds = create_optimization_bounds(
             schema=schema,
-            continuous_bounds_map=config.continuous_bounds_map,
+            continuous_bounds_map=continuous_bounds,
             start_at_zero=self.discretize_start_at_zero
         )
         self.lower_bounds = list(bounds[0])

{dragon_ml_toolbox-19.9.0 → dragon_ml_toolbox-19.11.0}/ml_tools/_core/_ML_utilities.py

@@ -46,17 +46,24 @@ class ArtifactFinder:
         └── FeatureSchema.json     (Required if `load_schema` is True)
     ```
     """
-    def __init__(self, directory: Union[str, Path], load_scaler: bool, load_schema: bool) -> None:
+    def __init__(self, 
+                 directory: Union[str, Path], 
+                 load_scaler: bool, 
+                 load_schema: bool,
+                 strict: bool=False,
+                 verbose: bool=True) -> None:
         """
         Args:
             directory (str | Path): The path to the directory that contains training artifacts.
             load_scaler (bool): If True, requires and searches for a scaler file `scaler_*.pth`.
             load_schema (bool): If True, requires and searches for a FeatureSchema file `FeatureSchema.json`.
+            strict (bool): If True, raises an error if any artifact is missing. If False, returns None for missing artifacts silently.
+            verbose (bool): Displays the missing artifacts in the directory or a success message.
         """
         # validate directory
         dir_path = make_fullpath(directory, enforce="directory")
         
-        parsing_dict = _find_model_artifacts(target_directory=dir_path, load_scaler=load_scaler, verbose=False)
+        parsing_dict = _find_model_artifacts(target_directory=dir_path, load_scaler=load_scaler, verbose=False, strict=strict)
         
         self._weights_path = parsing_dict[PytorchArtifactPathKeys.WEIGHTS_PATH]
         self._feature_names_path = parsing_dict[PytorchArtifactPathKeys.FEATURES_PATH]
@@ -64,71 +71,121 @@ class ArtifactFinder:
         self._model_architecture_path = parsing_dict[PytorchArtifactPathKeys.ARCHITECTURE_PATH]
         self._scaler_path = None
         self._schema = None
+        self._strict = strict
         
         if load_scaler:
             self._scaler_path = parsing_dict[PytorchArtifactPathKeys.SCALER_PATH]
             
         if load_schema:
-            self._schema = FeatureSchema.from_json(directory=dir_path)
+            try:
+                self._schema = FeatureSchema.from_json(directory=dir_path)
+            except Exception:
+                if strict:
+                    # FeatureSchema logs its own error details
+                    # _LOGGER.error(f"Failed to load FeatureSchema from '{dir_path.name}': {e}")
+                    raise FileNotFoundError()
+                else:
+                    # _LOGGER.warning(f"Could not load FeatureSchema from '{dir_path.name}': {e}")
+                    self._schema = None
+
+        # Process feature names
+        if self._feature_names_path is not None:
+            self._feature_names = self._process_text(self._feature_names_path)
+        else:
+            self._feature_names = None
+        # Process target names
+        if self._target_names_path is not None:
+            self._target_names = self._process_text(self._target_names_path)
+        else:
+            self._target_names = None
+            
+        if verbose:
+            # log missing artifacts
+            missing_artifacts = []
+            if self._feature_names is None:
+                missing_artifacts.append("Feature Names")
+            if self._target_names is None:
+                missing_artifacts.append("Target Names")
+            if self._weights_path is None:
+                missing_artifacts.append("Weights File")
+            if self._model_architecture_path is None:
+                missing_artifacts.append("Model Architecture File")
+            if load_scaler and self._scaler_path is None:
+                missing_artifacts.append("Scaler File")
+            if load_schema and self._schema is None:
+                missing_artifacts.append("FeatureSchema File")
+            
+            if missing_artifacts:
+                _LOGGER.warning(f"Missing artifacts in '{dir_path.name}': {', '.join(missing_artifacts)}.")
+            else:
+                _LOGGER.info(f"All artifacts successfully loaded from '{dir_path.name}'.")
 
-        # Process text files
-        self._feature_names = self._process_text(self._feature_names_path)
-        self._target_names = self._process_text(self._target_names_path)
-        
     def _process_text(self, text_file_path: Path):
         list_strings = load_list_strings(text_file=text_file_path, verbose=False)
         return list_strings
     
     @property
-    def feature_names(self) -> list[str]:
+    def feature_names(self) -> Union[list[str], None]:
         """Returns the feature names as a list of strings."""
+        if self._strict and not self._feature_names:
+            _LOGGER.error("No feature names loaded for Strict mode.")
+            raise ValueError()
         return self._feature_names
     
     @property
-    def target_names(self) -> list[str]:
+    def target_names(self) -> Union[list[str], None]:
         """Returns the target names as a list of strings."""
+        if self._strict and not self._target_names:
+            _LOGGER.error("No target names loaded for Strict mode.")
+            raise ValueError()
         return self._target_names
     
     @property
-    def weights_path(self) -> Path:
+    def weights_path(self) -> Union[Path, None]:
         """Returns the path to the state dictionary pth file."""
+        if self._strict and self._weights_path is None:
+            _LOGGER.error("No weights file loaded for Strict mode.")
+            raise ValueError()
         return self._weights_path
     
     @property
-    def model_architecture_path(self) -> Path:
+    def model_architecture_path(self) -> Union[Path, None]:
         """Returns the path to the model architecture json file."""
+        if self._strict and self._model_architecture_path is None:
+            _LOGGER.error("No model architecture file loaded for Strict mode.")
+            raise ValueError()
         return self._model_architecture_path
     
     @property
-    def scaler_path(self) -> Path:
+    def scaler_path(self) -> Union[Path, None]:
         """Returns the path to the scaler file."""
-        if self._scaler_path is None:
-            _LOGGER.error("No scaler file loaded. Set 'load_scaler=True'.")
+        if self._strict and self._scaler_path is None:
+            _LOGGER.error("No scaler file loaded for Strict mode.")
             raise ValueError()
         else:
             return self._scaler_path
         
     @property
-    def feature_schema(self) -> FeatureSchema:
+    def feature_schema(self) -> Union[FeatureSchema, None]:
         """Returns the FeatureSchema object."""
-        if self._schema is None:
-            _LOGGER.error("No FeatureSchema loaded. Set 'load_schema=True'.")
+        if self._strict and self._schema is None:
+            _LOGGER.error("No FeatureSchema loaded for Strict mode.")
             raise ValueError()
         else:
             return self._schema
         
     def __repr__(self) -> str:
-        dir_name = self._weights_path.parent.name
-        n_features = len(self._feature_names)
-        n_targets = len(self._target_names)
+        dir_name = self._weights_path.parent.name if self._weights_path else "Unknown"
+        n_features = len(self._feature_names) if self._feature_names else "None"
+        n_targets = len(self._target_names) if self._target_names else "None"
         scaler_status = self._scaler_path.name if self._scaler_path else "None"
         schema_status = "Loaded" if self._schema else "None"
         
         return (
             f"{self.__class__.__name__}\n"
             f"    directory='{dir_name}'\n"
-            f"    weights='{self._weights_path.name}'\n"
-            f"    architecture='{self._model_architecture_path.name}'\n"
+            f"    weights='{self._weights_path.name if self._weights_path else 'None'}'\n"
+            f"    architecture='{self._model_architecture_path.name if self._model_architecture_path else 'None'}'\n"
             f"    scaler='{scaler_status}'\n"
             f"    schema='{schema_status}'\n"
             f"    features={n_features}\n" 
@@ -136,7 +193,7 @@ class ArtifactFinder:
         )
 
 
-def _find_model_artifacts(target_directory: Union[str,Path], load_scaler: bool, verbose: bool=False) -> dict[str, Path]:
+def _find_model_artifacts(target_directory: Union[str,Path], load_scaler: bool, verbose: bool=True, strict:bool=True) -> dict[str, Union[Path, None]]:
     """
     Scans a directory to find paths to model weights, target names, feature names, and model architecture. Optionally an scaler path if `load_scaler` is True.
     
@@ -155,41 +212,70 @@ def _find_model_artifacts(target_directory: Union[str,Path], load_scaler: bool,
         target_directory (str | Path): The path to the directory that contains training artifacts.
         load_scaler (bool): If True, the function requires and searches for a scaler file `scaler_*.pth`.
         verbose (bool): If True, enables detailed logging during the search process.
+        strict (bool): If True, raises errors on missing files. If False, returns None for missing files.
     """
     # validate directory
     dir_path = make_fullpath(target_directory, enforce="directory")
     dir_name = dir_path.name
     
     # find files
-    model_pth_dict = list_files_by_extension(directory=dir_path, extension="pth", verbose=verbose)
+    model_pth_dict = list_files_by_extension(directory=dir_path, extension="pth", verbose=False, raise_on_empty=False)
     
-    # restriction
-    if load_scaler:
-        if len(model_pth_dict) != 2:
-            _LOGGER.error(f"Directory '{dir_name}' should contain exactly 2 '.pth' files: scaler and weights.")
-            raise IOError()
-    else:
-        if len(model_pth_dict) != 1:
-            _LOGGER.error(f"Directory '{dir_name}' should contain exactly 1 '.pth' file for weights.")
+    if not model_pth_dict:
+        pth_msg=f"No '.pth' files found in directory: {dir_name}."
+        if strict:
+            _LOGGER.error(pth_msg)
             raise IOError()
+        else:
+            if verbose:
+                _LOGGER.warning(pth_msg)
+            model_pth_dict = None
+    
+    # restriction
+    if model_pth_dict is not None:
+        valid_count = False
+        msg = ""
+        
+        if load_scaler:
+            if len(model_pth_dict) == 2:
+                valid_count = True
+            else:
+                msg = f"Directory '{dir_name}' should contain exactly 2 '.pth' files: scaler and weights. Found {len(model_pth_dict)}."
+        else:
+            if len(model_pth_dict) == 1:
+                valid_count = True
+            else:
+                msg = f"Directory '{dir_name}' should contain exactly 1 '.pth' file for weights. Found {len(model_pth_dict)}."
+        
+        # Respect strict mode for count mismatch
+        if not valid_count:
+            if strict:
+                _LOGGER.error(msg)
+                raise IOError()
+            else:
+                if verbose:
+                    _LOGGER.warning(msg)
+                # Invalidate dictionary
+                model_pth_dict = None
     
     ##### Scaler and Weights #####
     scaler_path = None
     weights_path = None
     
     # load weights and scaler if present
-    for pth_filename, pth_path in model_pth_dict.items():
-        if load_scaler and pth_filename.lower().startswith(DatasetKeys.SCALER_PREFIX):
-            scaler_path = pth_path
-        else:
-            weights_path = pth_path
+    if model_pth_dict is not None:
+        for pth_filename, pth_path in model_pth_dict.items():
+            if load_scaler and pth_filename.lower().startswith(DatasetKeys.SCALER_PREFIX):
+                scaler_path = pth_path
+            else:
+                weights_path = pth_path
     
     # validation
-    if not weights_path:
+    if not weights_path and strict:
         _LOGGER.error(f"Error parsing the model weights path from '{dir_name}'")
         raise IOError()
     
-    if load_scaler and not scaler_path:
+    if strict and load_scaler and not scaler_path:
         _LOGGER.error(f"Error parsing the scaler path from '{dir_name}'")
         raise IOError()
     
@@ -198,32 +284,44 @@ def _find_model_artifacts(target_directory: Union[str,Path], load_scaler: bool,
     feature_names_path = None
     
     # load feature and target names
-    model_txt_dict = list_files_by_extension(directory=dir_path, extension="txt", verbose=verbose)
+    model_txt_dict = list_files_by_extension(directory=dir_path, extension="txt", verbose=False, raise_on_empty=False)
     
+    # if the directory has no txt files, the loop is skipped
     for txt_filename, txt_path in model_txt_dict.items():
         if txt_filename == DatasetKeys.FEATURE_NAMES:
             feature_names_path = txt_path
         elif txt_filename == DatasetKeys.TARGET_NAMES:
             target_names_path = txt_path
     
-    # validation
-    if not target_names_path or not feature_names_path:
-        _LOGGER.error(f"Error parsing features path or targets path from '{dir_name}'")
+    # validation per case
+    if strict and not target_names_path:
+        _LOGGER.error(f"Error parsing the target names path from '{dir_name}'")
         raise IOError()
+    elif verbose and not target_names_path:
+        _LOGGER.warning(f"Target names file not found in '{dir_name}'.")
     
+    if strict and not feature_names_path:
+        _LOGGER.error(f"Error parsing the feature names path from '{dir_name}'")
+        raise IOError()
+    elif verbose and not feature_names_path:
+        _LOGGER.warning(f"Feature names file not found in '{dir_name}'.")
+
     ##### load model architecture path #####
     architecture_path = None
     
-    model_json_dict = list_files_by_extension(directory=dir_path, extension="json", verbose=verbose)
+    model_json_dict = list_files_by_extension(directory=dir_path, extension="json", verbose=False, raise_on_empty=False)
     
+    # if the directory has no json files, the loop is skipped
     for json_filename, json_path in model_json_dict.items():
         if json_filename == PytorchModelArchitectureKeys.SAVENAME:
             architecture_path = json_path
     
     # validation
-    if not architecture_path:
+    if strict and not architecture_path:
         _LOGGER.error(f"Error parsing the model architecture path from '{dir_name}'")
         raise IOError()
+    elif verbose and not architecture_path:
+        _LOGGER.warning(f"Model architecture file not found in '{dir_name}'.")
     
     ##### Paths dictionary #####
     parsing_dict = {
@@ -233,7 +331,7 @@ def _find_model_artifacts(target_directory: Union[str,Path], load_scaler: bool,
         PytorchArtifactPathKeys.TARGETS_PATH: target_names_path,
     }
     
-    if scaler_path is not None:
+    if load_scaler:
         parsing_dict[PytorchArtifactPathKeys.SCALER_PATH] = scaler_path
     
     return parsing_dict
@@ -246,6 +344,9 @@ def find_model_artifacts_multi(target_directory: Union[str,Path], load_scaler: b
     This function operates on a specific directory structure. It expects the
     `target_directory` to contain one or more subdirectories, where each
     subdirectory represents a single trained model result.
+    
+    This function works using a strict mode, meaning that it will raise errors if
+    any required artifacts are missing in a model's subdirectory.
 
     The expected directory structure for each model is as follows:
     ```
@@ -278,14 +379,16 @@ def find_model_artifacts_multi(target_directory: Union[str,Path], load_scaler: b
     all_artifacts: list[dict[str, Path]] = list()
     
     # find model directories
-    result_dirs_dict = list_subdirectories(root_dir=root_path, verbose=verbose)
+    result_dirs_dict = list_subdirectories(root_dir=root_path, verbose=verbose, raise_on_empty=True)
     for _dir_name, dir_path in result_dirs_dict.items():
         
         parsing_dict = _find_model_artifacts(target_directory=dir_path,
                                             load_scaler=load_scaler,
-                                            verbose=verbose)
+                                            verbose=verbose,
+                                            strict=True)
         
-        all_artifacts.append(parsing_dict)
+        # parsing_dict is guaranteed to have all required paths due to strict=True
+        all_artifacts.append(parsing_dict)  # type: ignore
     
     return all_artifacts
 
@@ -721,7 +824,7 @@ def select_features_by_shap(
     root_path = make_fullpath(root_directory, enforce="directory")
 
     # --- Step 2: Directory and File Discovery ---
-    subdirectories = list_subdirectories(root_dir=root_path, verbose=False)
+    subdirectories = list_subdirectories(root_dir=root_path, verbose=False, raise_on_empty=True)
     
     shap_filename = SHAPKeys.SAVENAME + ".csv"