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

Fix bug with local importance plots (#71)

Author: sarah-segel

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).

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()