Merge pull request #113 from automl/bug/local-importance-plots

Fix bug with local importance plots (#71)
automl · Feb 21, 2024 · bff04ce · bff04ce
2 parents e6c1f2e + a0ac677
commit bff04ce
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -13,6 +13,7 @@
 - Disable debug mode.
 
 ## Bug-Fixes
+- Fix missing objective specification in LPI evaluator (#71).
 - Don't convert BOHB runs with status 'running' (consistent with SMAC).
 - Fix api examples (#68).
 - Reset inputs to fix error when subsequently selecting runs with different configspaces, objectives or budgets (#106).

diff --git a/deepcave/evaluators/lpi.py b/deepcave/evaluators/lpi.py
@@ -60,7 +60,9 @@ def calculate(
         self.rs = np.random.RandomState(seed)
 
         # Get data
-        df = self.run.get_encoded_data(budget=budget, specific=True, include_combined_cost=True)
+        df = self.run.get_encoded_data(
+            objectives=objectives, budget=budget, specific=True, include_combined_cost=True
+        )
         X = df[self.hp_names].to_numpy()
         Y = df[COMBINED_COST_NAME].to_numpy()