parsing/src/benepar/nkutil.py

class HParams:
    _skip_keys = ["populate_arguments", "set_from_args", "print", "to_dict"]

    def __init__(self, **kwargs):
        for k, v in kwargs.items():
            setattr(self, k, v)

    def __getitem__(self, item):
        return getattr(self, item)

    def __setitem__(self, item, value):
        if not hasattr(self, item):
            raise KeyError(f"Hyperparameter {item} has not been declared yet")
        setattr(self, item, value)

    def to_dict(self):
        res = {}
        for k in dir(self):
            if k.startswith("_") or k in self._skip_keys:
                continue
            res[k] = self[k]
        return res

    def populate_arguments(self, parser):
        for k in dir(self):
            if k.startswith("_") or k in self._skip_keys:
                continue
            v = self[k]
            k = k.replace("_", "-")
            if type(v) in (int, float, str):
                parser.add_argument(f"--{k}", type=type(v), default=v)
            elif isinstance(v, bool):
                if not v:
                    parser.add_argument(f"--{k}", action="store_true")
                else:
                    parser.add_argument(f"--no-{k}", action="store_false")

    def set_from_args(self, args):
        for k in dir(self):
            if k.startswith("_") or k in self._skip_keys:
                continue
            if hasattr(args, k):
                self[k] = getattr(args, k)
            elif hasattr(args, f"no_{k}"):
                self[k] = getattr(args, f"no_{k}")

    def print(self):
        for k in dir(self):
            if k.startswith("_") or k in self._skip_keys:
                continue
            print(k, repr(self[k]))