metric conditioned on multi

wandb · Aug 1, 2023 · 312de6e · 312de6e
1 parent 2487be2
commit 312de6e
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/jobs/sweep_schedulers/optuna_scheduler/optuna_scheduler.py b/jobs/sweep_schedulers/optuna_scheduler/optuna_scheduler.py
@@ -534,7 +534,10 @@ def _get_run_history(self, run_id: str) -> List[int]:
         history = api_run.scan_history(keys=names + ["_step"])
         metrics = []
         for log in history:
-            metrics += [tuple(log.get(key) for key in names)]
+            if self.is_multi_objective:
+                metrics += [tuple(log.get(key) for key in names)]
+            else:
+                metrics += [log.get(names[0])]
 
         if len(metrics) == 0 and api_run.lastHistoryStep > -1:
             logger.debug("No metrics, but lastHistoryStep exists")