fix a CI bug

microsoft · Sep 11, 2024 · 3695a7b · 3695a7b
1 parent ffb2ff5
commit 3695a7b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/rdagent/components/benchmark/eval_method.py b/rdagent/components/benchmark/eval_method.py
@@ -19,7 +19,7 @@
 from rdagent.components.coder.factor_coder.factor import FactorFBWorkspace
 from rdagent.core.conf import RD_AGENT_SETTINGS
 from rdagent.core.developer import Developer
-from rdagent.core.exception import CoderError
+from rdagent.core.exception import CoderError, CustomRunnerError
 from rdagent.core.experiment import Task, Workspace
 from rdagent.core.scenario import Scenario
 from rdagent.core.utils import multiprocessing_wrapper
@@ -32,7 +32,7 @@
 
 EVAL_RES = Dict[
     str,
-    List[Tuple[FactorEvaluator, Union[object, RunnerException]]],
+    List[Tuple[FactorEvaluator, Union[object, CustomRunnerError]]],
 ]