Bug fixing

automl · Aug 29, 2024 · cf2bfaa · cf2bfaa
1 parent 344bc12
commit cf2bfaa
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/deepcave/plugins/hyperparameter/ablation_paths.py b/deepcave/plugins/hyperparameter/ablation_paths.py
@@ -318,7 +318,7 @@ def process(run: AbstractRun, inputs: Dict[str, Any]) -> Dict[str, Any]:
         RuntimeError
             If the number of trees is not specified.
         """
-        objective: Union[Optional[Objective], list[Optional[Objective]], None] = None
+        objective: Union[Objective, list[Optional[Objective]], None] = None
         objective = run.get_objective(inputs["objective_id1"])
         if inputs["objective_id2"] not in (None, -1):
             objective = [objective, run.get_objective(inputs["objective_id2"])]

diff --git a/deepcave/plugins/hyperparameter/importances.py b/deepcave/plugins/hyperparameter/importances.py
@@ -342,7 +342,7 @@ def process(run: AbstractRun, inputs: Dict[str, Any]) -> Dict[str, Any]:
             If the number of trees is not specified.
             If the method is not found.
         """
-        objective: Union[Optional[Objective], list[Optional[Objective]], None] = None
+        objective: Union[Objective, list[Optional[Objective]], None] = None
         objective = run.get_objective(inputs["objective_id1"])
         if inputs["objective_id2"] not in (None, -1):
             objective = [objective, run.get_objective(inputs["objective_id2"])]