Skip to content

Commit ec70ba4

Browse files
committed
fix
Signed-off-by: Hemil Desai <[email protected]>
1 parent 6236229 commit ec70ba4

File tree

1 file changed

+9
-1
lines changed

1 file changed

+9
-1
lines changed

nemo_run/run/ray/slurm.py

Lines changed: 9 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -664,7 +664,11 @@ def run(self):
664664
]
665665
)
666666

667-
jump_arg_str = f"{executor.tunnel.user}@{executor.tunnel.host}"
667+
jump_arg_str = (
668+
f"{executor.tunnel.user}@{executor.tunnel.host}"
669+
if isinstance(executor.tunnel, SSHTunnel)
670+
else None
671+
)
668672
raw_jump_identity = getattr(executor.tunnel, "identity", None)
669673
jump_identity_path_for_proxy = None
670674
if raw_jump_identity:
@@ -1135,6 +1139,8 @@ def start(
11351139
remote_workdir,
11361140
],
11371141
check=True,
1142+
stdout=subprocess.DEVNULL,
1143+
stderr=subprocess.DEVNULL,
11381144
)
11391145
elif self.executor.packager is not None:
11401146
# Use the packager to create an archive which we then extract on the
@@ -1190,6 +1196,8 @@ def start(
11901196
remote_workdir,
11911197
],
11921198
check=True,
1199+
stdout=subprocess.DEVNULL,
1200+
stderr=subprocess.DEVNULL,
11931201
)
11941202

11951203
assert remote_workdir is not None, "workdir could not be determined"

0 commit comments

Comments
 (0)