def generate_dcgm_command()

in src/gpu_healthcheck/gpu_healthcheck.py [0:0]

• 7 lines of code
• 3 McCabe index (conditional complexity)

def generate_dcgm_command() -> str:
  """Generates the command to run DCGM diagnostic tool."""
  r_level = os.environ.get("R_LEVEL", "1")
  command = DCGM_COMMAND % r_level
  dcgm_params = os.environ.get("DCGM_PARAMS")
  if dcgm_params is not None and dcgm_params:
    command += " -p " + dcgm_params
  return command