Skip to content

Commit

Permalink
Fix style
Browse files Browse the repository at this point in the history
  • Loading branch information
KyleFromNVIDIA committed Mar 13, 2024
1 parent 981fa76 commit 2829cf4
Showing 1 changed file with 12 additions and 4 deletions.
16 changes: 12 additions & 4 deletions rapids_build_backend/config.py
Original file line number Diff line number Diff line change
Expand Up @@ -32,7 +32,9 @@ def __init__(self, dirname=".", config_settings=None):
def __getattr__(self, name):
config_name = name.replace("_", "-")
if config_name in Config.config_options:
default_value, allows_override, allowed_values = Config.config_options[config_name]
default_value, allows_override, allowed_values = Config.config_options[
config_name
]

# If overrides are allowed environment variables take precedence over the
# config_settings dict.
Expand All @@ -48,15 +50,21 @@ def __getattr__(self, name):
f"{env_var} must be 'true' or 'false', not {str_val}"
)
return str_val == "true"
return self._check_value(env_var, os.environ[env_var], allowed_values)
return self._check_value(
env_var, os.environ[env_var], allowed_values
)

if config_name in self.config_settings:
if isinstance(default_value, bool):
return self.config_settings[config_name] == "true"
return self._check_value(config_name, self.config_settings[config_name], allowed_values)
return self._check_value(
config_name, self.config_settings[config_name], allowed_values
)

try:
return self._check_value(config_name, self.config[config_name], allowed_values)
return self._check_value(
config_name, self.config[config_name], allowed_values
)
except KeyError:
if default_value is not None:
return default_value
Expand Down

0 comments on commit 2829cf4

Please sign in to comment.