def ensure_env_variables()

in src/neper_healthcheck/neper_runner.py [0:0]


def ensure_env_variables() -> None:
  """Ensure necessary environment variables are set."""
  required_envs = [
      "NODE_NAME",
      "NODE_IP",
      "GOOD_THROUGHPUT",
      "HEALTH_VALIDITY_HOURS",
      "POD_NAME",
      "JOB_NAME",
      "SERVICE_NAME",
      "DRY_RUN",
  ]
  for env in required_envs:
    if env not in os.environ:
      raise ValueError(f"Must set {env}")
    print("ENV %s=%s" % (env, os.environ[env]))