updated to raise proper error
This commit is contained in:
@@ -11,4 +11,5 @@ def check_env(
|
||||
assert all(isinstance(elem, str) for elem in var_list)
|
||||
# check that env vars are set
|
||||
for env_var in var_list:
|
||||
assert env_var in os.environ, f"environment variable not set: {env_var}"
|
||||
if env_var not in os.environ:
|
||||
raise OSError(f"environment variable not set: {env_var}")
|
||||
|
||||
Reference in New Issue
Block a user