Skip to content

Commit e24c10f

Browse files
committed
Fix
Signed-off-by: Hemil Desai <[email protected]>
1 parent f217264 commit e24c10f

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

nemo_run/run/torchx_backend/packaging.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -230,9 +230,9 @@ def _get_details_from_script(fn_or_script: Script, serialize_configs: bool):
230230

231231
if metadata:
232232
if USE_WITH_RAY_CLUSTER_KEY in metadata:
233-
assert isinstance(executor, SlurmExecutor), (
234-
f"{USE_WITH_RAY_CLUSTER_KEY} is only supported for SlurmExecutor"
235-
)
233+
assert isinstance(
234+
executor, SlurmExecutor
235+
), f"{USE_WITH_RAY_CLUSTER_KEY} is only supported for SlurmExecutor"
236236

237237
app_def.metadata = metadata
238238
return app_def
@@ -246,7 +246,7 @@ def merge_executables(app_defs: Iterator[specs.AppDef], name: str) -> specs.AppD
246246
if USE_WITH_RAY_CLUSTER_KEY in metadata:
247247
assert idx == 0, f"{USE_WITH_RAY_CLUSTER_KEY} is only supported for the first command"
248248

249-
result.metadata.update(app_def.metadata)
249+
result.metadata.update(metadata)
250250
result.roles.extend(app_def.roles)
251251
return result
252252

0 commit comments

Comments
 (0)