SIENTIAPDE-1579: Fix SientiaMlException propagation in ModelServing.search_runs_by_name to correctly raise the exception with its message, resolving a TypeError. Enhance training test script to support scenario-specific CSV files for different test cases. Update search_runs_by_name return type hint and apply minor code formatting.

model_manager/sientia/model_serving.py

@@ -56,7 +56,7 @@ class ModelServing:
     # Function to list runs for a given experiment
     def search_runs_by_name(
         self, experiment_names: list[str], order_by: None | list[str] = None
-    ) -> pd.DataFrame:
+    ) -> pd.DataFrame | list:
         """
         List runs for a specified MLflow experiment.
 
@@ -64,7 +64,7 @@ class ModelServing:
             experiment_names (list[str]): List with experiment_names to retrieve runs from.
 
         Returns:
-            pandas.DataFrame: A DataFrame containing run information.
+            Union[pd.DataFrame, list]: A DataFrame or list containing run information.
 
         Raises:
             SientiaMlException: If unable to search runs.
@@ -73,7 +73,7 @@ class ModelServing:
             runs = mlflow.search_runs(experiment_names=experiment_names, order_by=order_by)
         except SientiaMlException as e:
             logging.error(e)
-            raise SientiaMlException from e
+            raise SientiaMlException(str(e)) from e
         return runs
 
     def set_experiment(self, experiment_identifier: str) -> None:

model_manager/sientia/models.py

@@ -35,9 +35,7 @@ def validate_frontend_date_format(fmt: str | None) -> None:
         return
     if fmt not in ALLOWED_FRONTEND_DATE_FORMATS:
         allowed = ', '.join(sorted(ALLOWED_FRONTEND_DATE_FORMATS))
-        raise ValueError(
-            f'Invalid date_format "{fmt}". Allowed formats: {allowed}'
-        )
+        raise ValueError(f'Invalid date_format "{fmt}". Allowed formats: {allowed}')
 
 
 def _frontend_date_format_to_strftime(fmt: str | None) -> str | None: