From 5d93d25b9ee4c92183987b8b703dd73ea9f5c128 Mon Sep 17 00:00:00 2001 From: Youngeun Kwon Date: Thu, 8 May 2025 19:26:48 -0700 Subject: [PATCH] add sbatch network arg Signed-off-by: Youngeun Kwon --- nemo_run/core/execution/slurm.py | 1 + 1 file changed, 1 insertion(+) diff --git a/nemo_run/core/execution/slurm.py b/nemo_run/core/execution/slurm.py index 294cdf1a..6f436bdf 100644 --- a/nemo_run/core/execution/slurm.py +++ b/nemo_run/core/execution/slurm.py @@ -340,6 +340,7 @@ class ResourceRequest: setup_lines: Optional[str] = None het_group_indices: Optional[list[int]] = None segment: Optional[int] = None + network: Optional[str] = None #: Set by the executor; cannot be initialized job_name: str = field(init=False, default="nemo-job")