From 6d352fe887ab049a0933d1d78db8aeaf96a9814a Mon Sep 17 00:00:00 2001 From: Hemil Desai Date: Mon, 26 Aug 2024 12:33:19 -0700 Subject: [PATCH] Fix typo in SlurmExecutor docstring Signed-off-by: Hemil Desai --- src/nemo_run/core/execution/slurm.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/src/nemo_run/core/execution/slurm.py b/src/nemo_run/core/execution/slurm.py index 5fc522c..a8e74f9 100644 --- a/src/nemo_run/core/execution/slurm.py +++ b/src/nemo_run/core/execution/slurm.py @@ -69,7 +69,8 @@ def your_slurm_executor() -> run.SlurmExecutor: user=os.environ["SLURM_USER"], job_dir=os.environ["SLURM_JOBDIR"], ) - packager = GitArchivePackager(launcher="torchrun") + packager = GitArchivePackager() + launcher = "torchrun" executor = SlurmExecutor( account=os.environ["SLURM_ACCT"], partition=os.environ["SLURM_PARTITION"], @@ -79,6 +80,7 @@ def your_slurm_executor() -> run.SlurmExecutor: container_image=os.environ["BASE_IMAGE"], time="00:30:00", packager=packager, + launcher=launcher, ) return executor