class_docstr: This class is for testing via an external script - can do multiple gpus
class_docstr_tok: ["This","class","is","for","testing","via","an","external","script","-","can","do","multiple","gpus"]
code: def get_task_cmd(self, task, stage):
      # return a ready to run train cmd
      if task not in task_cmds:
          raise ValueError(
              f"don't know of task {task}, have {task_cmds.keys()}"
          )
  
      cmd = task_cmds[task]
      args_ds = f"--deepspeed {self.test_file_dir_str}/ds_config_{stage}.json".split()
  
      output_dir = self.get_auto_remove_tmp_dir()
      args_out = f"--output_dir {output_dir}".split()
  
      cmd += args_ds + args_out
  
      return cmd, output_dir
  
file_path: transformers/tests/deepspeed/test_model_zoo.py
imports: import itertools
  import os
  import subprocess
  from os.path import dirname
  from parameterized import parameterized
  from tests.trainer.test_trainer import TrainerIntegrationCommon
  from transformers import is_torch_available
  from transformers.testing_utils import TestCasePlus, execute_subprocess_async, get_gpu_count, get_tests_dir, require_deepspeed, require_torch_gpu, slow
  from transformers.trainer_utils import set_seed
label_desc: Use image node_id 1 for calling the TestDeepSpeedModelZoo obj's underlying member method code with example usage: obj.get_task_cmd(task, stage) and returns: cmd,  output_dir