SIENTIAPDE-1312
Refactor error handling in MLFlowRepository and update tests - Improved error message formatting in MLFlowRepository for better readability. - Updated test assertions to ensure correct calls to MLflow methods during experiment retrieval and creation.
This commit is contained in:
@@ -178,7 +178,9 @@ class MLFlowRepository:
|
||||
experiment_id = mlflow.create_experiment(experiment_name)
|
||||
experiment = mlflow.get_experiment(experiment_id)
|
||||
if experiment is None:
|
||||
raise ValueError(f'Experiment {experiment_name} not found after creation, unknown reason')
|
||||
raise ValueError(
|
||||
f'Experiment {experiment_name} not found after creation, unknown reason'
|
||||
)
|
||||
else:
|
||||
raise ValueError(f'Experiment {experiment_name} not found')
|
||||
|
||||
|
||||
Reference in New Issue
Block a user