diff --git a/mlos_bench/mlos_bench/optimizers/mlos_core_optimizer.py b/mlos_bench/mlos_bench/optimizers/mlos_core_optimizer.py index 5d225ed8a5..649b070123 100644 --- a/mlos_bench/mlos_bench/optimizers/mlos_core_optimizer.py +++ b/mlos_bench/mlos_bench/optimizers/mlos_core_optimizer.py @@ -65,9 +65,10 @@ def __init__( # Make sure max_trials >= max_suggestions. if "max_trials" not in self._config: self._config["max_trials"] = self._max_suggestions - assert ( - int(self._config["max_trials"]) >= self._max_suggestions - ), f"max_trials {self._config.get('max_trials')} <= max_suggestions {self._max_suggestions}" + assert int(self._config["max_trials"]) >= self._max_suggestions, ( + f"max_trials {self._config.get('max_trials')} " + f"<= max_suggestions{self._max_suggestions}" + ) if "run_name" not in self._config and self.experiment_id: self._config["run_name"] = self.experiment_id