Bug fixing

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

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/deepcave/plugins/hyperparameter/ablation_paths.py b/deepcave/plugins/hyperparameter/ablation_paths.py
@@ -25,7 +25,7 @@
     - Ablation_Paths: This class provides a plugin for the visualization of the ablation paths.
 """
 
-from typing import Any, Callable, Dict, List, Union
+from typing import Any, Callable, Dict, List, Union, Optional
 
 import math
 from io import StringIO
@@ -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[Objective, list[Objective], None] = None
+        objective: Union[Optional[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[Objective, list[Objective], None] = None
+        objective: Union[Optional[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"])]