Fix logging for packaging jobs in Slurm executor (#160)

hemildesai · web-flow · commit 4d056535b5cc · 2025-02-24T10:12:38.000-08:00
Signed-off-by: Hemil Desai &lt;hemild@nvidia.com&gt;
diff --git a/src/nemo_run/core/execution/slurm.py b/src/nemo_run/core/execution/slurm.py
@@ -567,7 +567,7 @@ def package_configs(self, *cfgs: tuple[str, str]) -> list[str]:
     def package(self, packager: Packager, job_name: str):
         if job_name in self.tunnel.packaging_jobs and not packager.symlink_from_remote_dir:
             logger.info(
-                f"Packaging for job {job_name} in tunnel {self.tunnel} already done. Skipping subsequent packagings.\n"
+                f"Packaging for job {job_name} in tunnel {self.tunnel.key} already done. Skipping subsequent packagings.\n"
                 "This may cause issues if you have multiple tasks with the same name but different packagers, as only the first packager will be used."
             )
             return

Original file line number	Diff line number	Diff line change
`@@ -567,7 +567,7 @@ def package_configs(self, *cfgs: tuple[str, str]) -> list[str]:`
`567`	`567`	`def package(self, packager: Packager, job_name: str):`
`568`	`568`	`if job_name in self.tunnel.packaging_jobs and not packager.symlink_from_remote_dir:`
`569`	`569`	`logger.info(`
`570`		`- f"Packaging for job {job_name} in tunnel {self.tunnel} already done. Skipping subsequent packagings.\n"`
	`570`	`+ f"Packaging for job {job_name} in tunnel {self.tunnel.key} already done. Skipping subsequent packagings.\n"`
`571`	`571`	`"This may cause issues if you have multiple tasks with the same name but different packagers, as only the first packager will be used."`
`572`	`572`	`)`
`573`	`573`	`return`