o wOi@sddlZdedefddZdS)Nenv_namereturncCs.tj|d}|durd|d}t||S)z Tries to retrieve environment variable. Raises ``ValueError`` if no environment variable found. Args: env_name (str): Name of the env variable NzEnvironment variable z expected, but not set)osenvironget ValueError)rvaluemsgr J/home/ubuntu/.local/lib/python3.10/site-packages/torchelastic/utils/api.pyget_env_variable_or_raise s  r )rstrr r r r r s