Skip to content

Commit f2cbd72

Browse files
add env variable for ray port
1 parent 04eddfe commit f2cbd72

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

src/agentlab/experiments/launch_exp.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -9,6 +9,7 @@
99
from agentlab.experiments.loop import ExpArgs, yield_all_exp_results
1010

1111
RAY_PUBLIC_DASHBOARD = os.environ.get("RAY_PUBLIC_DASHBOARD", "false") == "true"
12+
RAY_DASHBOARD_PORT = os.environ.get("RAY_DASHBOARD_PORT")
1213

1314

1415
def run_experiments(
@@ -86,7 +87,9 @@ def run_experiments(
8687
from agentlab.experiments.graph_execution_ray import execute_task_graph, ray
8788

8889
ray.init(
89-
num_cpus=n_jobs, dashboard_host="0.0.0.0" if RAY_PUBLIC_DASHBOARD else "127.0.0.1"
90+
num_cpus=n_jobs,
91+
dashboard_host="0.0.0.0" if RAY_PUBLIC_DASHBOARD else "127.0.0.1",
92+
dashboard_port=None if RAY_DASHBOARD_PORT is None else int(RAY_DASHBOARD_PORT),
9093
)
9194
try:
9295
execute_task_graph(exp_args_list, avg_step_timeout=avg_step_timeout)

0 commit comments

Comments
 (0)