fix: record bugdet in runhistory

automl · May 16, 2024 · d46c636 · d46c636
1 parent 601adb4
commit d46c636
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/hydra_plugins/hypersweeper/hypersweeper_sweeper.py b/hydra_plugins/hypersweeper/hypersweeper_sweeper.py
@@ -350,7 +350,10 @@ def record_iteration(self, performances, configs, budgets):
         for i in range(len(configs)):
             self.history["configs"].append(configs[i])
             self.history["performances"].append(performances[i])
-            self.history["budgets"].append(budgets[i])
+            if budgets[i] is not None:
+                self.history["budgets"].append(budgets[i])
+            else:
+                self.history["budgets"].append(self.max_budget)
         self.iteration += 1
 
         if self.wandb_project: