File size: 1,526 Bytes
f6d1f07 af4ca53 f6d1f07 af4ca53 f6d1f07 bfa7f57 af4ca53 bfa7f57 f6d1f07 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 |
import os
from .logging_utils import get_logger
logger = get_logger()
class Settings:
_instance = None
_settings = {}
def __new__(cls):
if cls._instance is None:
cls._instance = super().__new__(cls)
return cls._instance
def __setattr__(self, key, value):
if key.endswith("_key") or key in {"_instance", "_settings"}:
raise AttributeError(f"Modifying '{key}' is not allowed.")
if key in self._settings:
logger.info(
f"unitxt.settings.{key} changed: {self._settings[key]} -> {value}"
)
self._settings[key] = value
def __getattr__(self, key):
if key.endswith("_key"):
actual_key = key[:-4] # Remove the "_key" suffix
return self.environment_variable_key_name(actual_key)
env_value = os.getenv(self.environment_variable_key_name(key))
if env_value is not None:
return env_value
if key in self._settings:
return self._settings[key]
raise AttributeError(f"'{key}' not found")
def environment_variable_key_name(self, key):
return "UNITXT_" + key.upper()
def get_all_environment_variables(self):
return [
self.environment_variable_key_name(key) for key in self._settings.keys()
]
settings = Settings()
settings.allow_unverified_code = False
settings.use_only_local_catalogs = False
settings.global_loader_limit = None
def get_settings():
return Settings()
|