Fix typo in SlurmExecutor docstring (#8)

Signed-off-by: Hemil Desai <hemild@nvidia.com>
NVIDIA · Aug 26, 2024 · 0001e70 · 0001e70
1 parent d614a8a
commit 0001e70
Showing 1 changed file with 3 additions and 1 deletion.
diff --git 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