Update tests.

Author: qiuosier

tests/unitary/default_setup/jobs/test_jobs_pytorch_ddp.py

@@ -137,12 +137,12 @@ def test_create_job_runs(self, patched_run, *args):
  [
  {
  "display_name": "None-0",
- "environment_variables": {"RANK": "0", "WORLD_SIZE": "2"},
+ "environment_variables": {"NODE_RANK": "0", "WORLD_SIZE": "2"},
  },
  {
  "display_name": "None-1",
  "environment_variables": {
- "RANK": "1",
+ "NODE_RANK": "1",
  "WORLD_SIZE": "2",
  "MAIN_JOB_RUN_OCID": test_ocid,
  },

tests/unitary/with_extras/jobs/test_pytorch_ddp.py

@@ -108,7 +108,7 @@ def test_install_deps(self, run_command):
  cmd_list,
  [
  "pip install -r abc/requirements.txt",
- "pip install abc==1.0",
+ "pip install 'abc==1.0'",
  ],
  )
 
@@ -172,7 +172,7 @@ def init_runner(self):
  driver.CONST_ENV_DEEPSPEED: "1",
  driver.OCI__WORKER_COUNT: "1",
  driver.CONST_ENV_LAUNCH_CMD: "accelerate launch train.py --data abc",
- "RANK": "0",
+ "NODE_RANK": "0",
  },
  )
  @mock.patch("ads.jobs.templates.driver_pytorch.DeepSpeedRunner.run_deepspeed_host")