Update test_jobs_pytorch_ddp.py

qiuosier · qiuosier · commit be6334d6ee8a · 2023-07-25T16:38:29.000-04:00
diff --git a/tests/unitary/default_setup/jobs/test_jobs_pytorch_ddp.py b/tests/unitary/default_setup/jobs/test_jobs_pytorch_ddp.py
@@ -137,13 +137,13 @@ def test_create_job_runs(self, patched_run, *args):
             [
                 {
                     "display_name": "None-0",
-                    "environment_variables": {"RANK": "0", "WORLD_SIZE": 2},
+                    "environment_variables": {"RANK": "0", "WORLD_SIZE": "2"},
                 },
                 {
                     "display_name": "None-1",
                     "environment_variables": {
                         "RANK": "1",
-                        "WORLD_SIZE": 2,
+                        "WORLD_SIZE": "2",
                         "MAIN_JOB_RUN_OCID": test_ocid,
                     },
                 },

Original file line number	Diff line number	Diff line change
`@@ -137,13 +137,13 @@ def test_create_job_runs(self, patched_run, *args):`
`137`	`137`	`[`
`138`	`138`	`{`
`139`	`139`	`"display_name": "None-0",`
`140`		`- "environment_variables": {"RANK": "0", "WORLD_SIZE": 2},`
	`140`	`+ "environment_variables": {"RANK": "0", "WORLD_SIZE": "2"},`
`141`	`141`	`},`
`142`	`142`	`{`
`143`	`143`	`"display_name": "None-1",`
`144`	`144`	`"environment_variables": {`
`145`	`145`	`"RANK": "1",`
`146`		`- "WORLD_SIZE": 2,`
	`146`	`+ "WORLD_SIZE": "2",`
`147`	`147`	`"MAIN_JOB_RUN_OCID": test_ocid,`
`148`	`148`	`},`
`149`	`149`	`},`