From 283c0a13ec4a3bf792dc09daa8b4d6358d6ebfbe Mon Sep 17 00:00:00 2001 From: Hemil Desai Date: Mon, 16 Dec 2024 11:52:36 -0800 Subject: [PATCH] Allow multiple docker experiments with same task names to run at the same time (#124) Signed-off-by: Hemil Desai --- src/nemo_run/run/torchx_backend/schedulers/docker.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/src/nemo_run/run/torchx_backend/schedulers/docker.py b/src/nemo_run/run/torchx_backend/schedulers/docker.py index 0f6bca2..8dfbaa5 100644 --- a/src/nemo_run/run/torchx_backend/schedulers/docker.py +++ b/src/nemo_run/run/torchx_backend/schedulers/docker.py @@ -88,7 +88,10 @@ def _submit_dryrun(self, app: AppDef, cfg: Executor) -> AppDryRunInfo[DockerJobR cmd = [role.entrypoint] + role.args containers.append( DockerContainer( - name=role.name, command=cmd, executor=_current_executor, extra_env=role.env + name=f"{executor.experiment_id}_{role.name}", + command=cmd, + executor=_current_executor, + extra_env=role.env, ) )