third_party/TopicFM/src/__init__.py

from yacs.config import CfgNode
from .config.default import _CN


def lower_config(yacs_cfg):
    if not isinstance(yacs_cfg, CfgNode):
        return yacs_cfg
    return {k.lower(): lower_config(v) for k, v in yacs_cfg.items()}


def get_model_cfg():
    cfg = lower_config(lower_config(_CN))
    return cfg["model"]