Merge branch 'fix-tabs' into fix-gradio

Author: recursix

src/agentlab/experiments/graph_execution_ray.py

@@ -2,12 +2,14 @@
 
 # # Disable Ray log deduplication
 # os.environ["RAY_DEDUP_LOGS"] = "0"
+import logging
 import time
-import ray
+
 import bgym
-from agentlab.experiments.exp_utils import run_exp, _episode_timeout
+import ray
 from ray.util import state
-import logging
+
+from agentlab.experiments.exp_utils import _episode_timeout, run_exp
 
 logger = logging.getLogger(__name__)
 
@@ -36,6 +38,7 @@ def get_task(exp_arg: bgym.ExpArgs):
         get_task(exp_arg)
 
     max_timeout = max([_episode_timeout(exp_args, avg_step_timeout) for exp_args in exp_args_list])
+
     return poll_for_timeout(task_map, max_timeout, poll_interval=max_timeout * 0.1)