code: def get_script(args, step_num):
      model_size = get_model_size(args, step_num)
      script = os.path.join(
          os.getcwd(),
          step_dirs[step_num],
          "training_scripts/opt/",
          args.deployment_type,
          f"run_{model_size}.sh",
      )
      assert os.path.isfile(
          script
      ), f"{script} does not exist.\n\n Use examples in {os.path.dirname(script)} as a template."
  
      return script
  
file_path: deepspeedexamples/applications/DeepSpeed-Chat/e2e_rlhf.py
imports: import argparse
  import warnings
  import subprocess
  import os
  import datetime
  import time
  import sys
label_desc: Use image node_id 5 for calling a global function with example usage: get_script(args, step_num) and returns: script