[fix] no magic numbers -- forgot the attribute

automl · Jun 5, 2024 · ee234cd · ee234cd
1 parent adf4155
commit ee234cd
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/carps/optimizers/synetune.py b/carps/optimizers/synetune.py
@@ -123,7 +123,7 @@ def __init__(
         self.fidelity_type: str = self.task.fidelity_type
         self.configspace = self.problem.configspace
         self.metric: str | list[str] = self.task.objectives
-
+        self.conversion_factor = conversion_factor
         self.trial_counter: int = 0
 
         self.optimizer_name = optimizer_name
@@ -356,6 +356,11 @@ def _setup_optimizer(self) -> SyneTrialScheduler:
             del self.optimizer_kwargs["metric"]
             del self.optimizer_kwargs["resource_attr"]
 
+        if self.optimizer_name in ['SyncMOBSTER']:
+            del self.optimizer_kwargs['metrics']
+            del self.optimizer_kwargs['time_attr']
+
+
         return optimizers_dict[self.optimizer_name](**self.optimizer_kwargs)
 
     def get_current_incumbent(self) -> Incumbent: