code: def run_shell_command(cmd, shell=False, env_vars={}):
      env = os.environ
      env = {**env, **env_vars}
      result = subprocess.run(
          cmd, shell=shell, capture_output=True, env=env
      )
      if result.returncode != 0:
          print("FAILED - {}".format(" ".join(cmd)))
          print(result.stderr.decode())
          # sys.exit(result.returncode)
      return (
          result.returncode,
          result.stderr.decode(),
          result.stdout.decode(),
      )
  
file_path: jax/build/rocm/run_single_gpu.py
imports: import argparse
  import os
  import re
  import subprocess
  import threading
  from concurrent.futures import ThreadPoolExecutor
label_desc: Use image node_id 1 for calling a global function with example usage: run_shell_command(cmd, shell, env_vars) and returns: result,  result,  result