def main()

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


def main() -> None:
  """entry point."""
  node_name = os.environ.get("NODE_NAME")

  print("diagnostic on vm '%s' is 'initiated'" % node_name)

  reboot_required = run_reboot_required_check(node_name)
  run_dcgm_diag(node_name, reboot_required)
  checker_common.add_label(
      node_name,
      HEALTHCHECK_TIME_LABEL_KEY,
      f"{int(time.time())}",
      K_ADD_LABEL_FORMAT,
  )