Bug fixing

automl · Aug 29, 2024 · 2ceab26 · 2ceab26
1 parent 515f2ca
commit 2ceab26
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/deepcave/evaluators/mo_fanova.py b/deepcave/evaluators/mo_fanova.py
@@ -124,7 +124,7 @@ def calculate(
         ).reset_index(drop=True)
 
     def get_importances(
-        self, hp_names: Optional[List[str]] = None, sort: bool = True
+            self, hp_names: Optional[List[str]] = None, depth: int = 1, sort: bool = True
     ) -> Union[Dict[Union[str, Tuple[str, ...]], Tuple[float, float, float, float]], str]:
         """
         Return the importance scores from the passed Hyperparameter names.

diff --git a/deepcave/plugins/hyperparameter/ablation_paths.py b/deepcave/plugins/hyperparameter/ablation_paths.py
@@ -340,7 +340,7 @@ def process(run: AbstractRun, inputs: Dict[str, Any]) -> Dict[str, Any]:
             assert isinstance(budget, (int, float))
             evaluator.calculate(objective, budget, n_trees=n_trees, seed=0)
             if isinstance(objective, list):
-                data[budget_id] = evaluator.get_importances(hp_names=None)
+                data[budget_id] = evaluator.get_importances()
             else:
                 performances = evaluator.get_ablation_performances()
                 improvements = evaluator.get_ablation_improvements()