diff --git a/marymorstan/pipeline.py b/marymorstan/pipeline.py index a0837e129f2bfb1c30118c4e31926bcbbdc2db6b..ea83a228580c7fd8e053e93f32883ad718aea3fd 100644 --- a/marymorstan/pipeline.py +++ b/marymorstan/pipeline.py @@ -594,7 +594,8 @@ class MLPipelineComponent(): for hyperparameter_name, hyperparameter_attributes in search_space_node['hyperparameters'].items(): - generated_value = generate_hyperparameter_value(hyperparameter_node=hyperparameter_attributes, use_default=use_default, random_state=random_state) + generated_value = generate_hyperparameter_value(hyperparameter_node=hyperparameter_attributes, + use_default=use_default, random_state=random_state) if generated_value is None: continue @@ -605,11 +606,12 @@ class MLPipelineComponent(): # TODO find a smart way to remove hyperparameters/values when a forbidden clause is raise - if not MLPipelineComponent._forbidden_hyperparameters_exists(search_space_node=search_space_node, hyperparameters=hyperparameters): + if not MLPipelineComponent._forbidden_hyperparameters_exists(search_space_node=search_space_node, + hyperparameters=hyperparameters): break if number_of_try >= limit_number_of_try: - message = (f"Impossible to generate a valid combination of hyperparamaters " + message = (f"Impossible to generate a valid combination of hyperparameters " f"in search space {search_space_node} after {number_of_try} tentatives") logging.error(message) raise InvalidHyperparameters(message)