.

ahmadsharif1 · ahmadsharif1 · commit 60e80d8ff1fd · 2025-08-24T15:26:02.000-07:00
diff --git a/torchx/schedulers/test/slurm_scheduler_test.py b/torchx/schedulers/test/slurm_scheduler_test.py
@@ -128,6 +128,7 @@ def test_replica_request(self, mock_version: MagicMock) -> None:
                 "--cpus-per-task=2",
                 "--mem=10",
                 "--gpus-per-task=3",
+                "--ntasks=1",
             ],
         )
         self.assertEqual(
@@ -163,6 +164,7 @@ def test_replica_request_nomem(self, mock_version: MagicMock) -> None:
                 "--ntasks-per-node=1",
                 "--cpus-per-task=2",
                 "--gpus-per-task=3",
+                "--ntasks=1",
             ],
         )
 

Original file line number	Diff line number	Diff line change
`@@ -128,6 +128,7 @@ def test_replica_request(self, mock_version: MagicMock) -> None:`
`128`	`128`	`"--cpus-per-task=2",`
`129`	`129`	`"--mem=10",`
`130`	`130`	`"--gpus-per-task=3",`
	`131`	`+ "--ntasks=1",`
`131`	`132`	`],`
`132`	`133`	`)`
`133`	`134`	`self.assertEqual(`
`@@ -163,6 +164,7 @@ def test_replica_request_nomem(self, mock_version: MagicMock) -> None:`
`163`	`164`	`"--ntasks-per-node=1",`
`164`	`165`	`"--cpus-per-task=2",`
`165`	`166`	`"--gpus-per-task=3",`
	`167`	`+ "--ntasks=1",`
`166`	`168`	`],`
`167`	`169`	`)`
`168`	`170`