"""pygram11 configuration."""
from typing import Dict, Iterator
config: Dict[str, int] = {
"thresholds.fix1d": 10_000,
"thresholds.fix1dmw": 10_000,
"thresholds.fix2d": 10_000,
"thresholds.var1d": 5_000,
"thresholds.var1dmw": 5_000,
"thresholds.var2d": 5_000,
}
[docs]def get(key: str) -> int:
"""Retrieve a configuration value given a key.
Parameters
----------
key : str
Desired configuration parameter key.
Returns
-------
int
Value of the key.
"""
global config
return config[key]
[docs]def set(key: str, value: int) -> None:
"""Set a configuration key's value.
Parameters
----------
key : str
Desired configuration parameter key to modify.
value : int
New value for the given key.
"""
global config
if key not in config:
raise ValueError(f"{key} not in config")
config[key] = value
def threshold_keys() -> Iterator[str]:
"""All available keys in the configuration dictionary."""
global config
for k in config.keys():
yield k