diff --git a/model_analyzer/config/generate/optuna_run_config_generator.py b/model_analyzer/config/generate/optuna_run_config_generator.py index 47356935..c53192f1 100755 --- a/model_analyzer/config/generate/optuna_run_config_generator.py +++ b/model_analyzer/config/generate/optuna_run_config_generator.py @@ -357,7 +357,7 @@ def _create_trial_objectives(self, trial: optuna.Trial) -> AllTrialObjectives: ) if parameter: - objective_name = objective_name = self._create_trial_objective_name( + objective_name = self._create_trial_objective_name( model_name=model_name, parameter_name=parameter_name ) diff --git a/tests/test_optuna_run_config_generator.py b/tests/test_optuna_run_config_generator.py index b51c2ce3..01953d1f 100755 --- a/tests/test_optuna_run_config_generator.py +++ b/tests/test_optuna_run_config_generator.py @@ -388,9 +388,6 @@ def test_create_run_multi_model_config(self): add_perf_config = run_config.model_run_configs()[0].perf_config() vgg_perf_config = run_config.model_run_configs()[0].perf_config() - # PA Config (Seed=100) - # ===================================================================== - # ADD_SUB + PA Config (Seed=100) # ===================================================================== self.assertEqual(add_model_config.to_dict()["name"], "add_sub")