def ensure_env_variables()

in src/nccl_healthcheck/nccl_startup.py [0:0]

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

def ensure_env_variables() -> None:
  """Ensure necessary environment variables are set."""
  required_envs = [
      "NODE_NAME",
      "NHOSTS",
      "nr",
      "JOB_COMPLETION_INDEX",
      "BANDWIDTH_THRESHOLD",
      "START_MESSAGE_SIZE",
      "END_MESSAGE_SIZE",
      "DRY_RUN",
      "INSTANCE_TYPE",
  ]
  for env in required_envs:
    if env not in os.environ:
      raise ValueError(f"Must set {env}")