displaying exp names in ray dashboard

TLSDC · TLSDC · commit abfded9351aa · 2024-11-14T13:53:28.000-05:00
diff --git a/src/agentlab/experiments/graph_execution_ray.py b/src/agentlab/experiments/graph_execution_ray.py
@@ -28,7 +28,7 @@ def get_task(exp_arg: bgym.ExpArgs):
             dependency_tasks = [get_task(exp_args_map[dep_key]) for dep_key in exp_arg.depends_on]
 
             # Create new task that depends on the dependency results
-            task_map[exp_arg.exp_id] = run_exp.remote(
+            task_map[exp_arg.exp_id] = run_exp.options(name=f"{exp_arg.exp_name}").remote(
                 exp_arg, *dependency_tasks, avg_step_timeout=avg_step_timeout
             )
         return task_map[exp_arg.exp_id]

Original file line number	Diff line number	Diff line change
`@@ -28,7 +28,7 @@ def get_task(exp_arg: bgym.ExpArgs):`
`28`	`28`	`dependency_tasks = [get_task(exp_args_map[dep_key]) for dep_key in exp_arg.depends_on]`
`29`	`29`
`30`	`30`	`# Create new task that depends on the dependency results`
`31`		`- task_map[exp_arg.exp_id] = run_exp.remote(`
	`31`	`+ task_map[exp_arg.exp_id] = run_exp.options(name=f"{exp_arg.exp_name}").remote(`
`32`	`32`	`exp_arg, *dependency_tasks, avg_step_timeout=avg_step_timeout`
`33`	`33`	`)`
`34`	`34`	`return task_map[exp_arg.exp_id]`