def get_hfconfig()

in docker_images/k2/app/common.py [0:0]

• 12 lines of code
• 4 McCabe index (conditional complexity)

def get_hfconfig(model_id, config_name="hf_demo"):
    info = HfApi().model_info(repo_id=model_id)
    config_file = hf_hub_download(model_id, filename="config.json")
    with open(config_file) as config:
        info.config = json.load(config)

    if info.config and config_name is not None:
        if config_name in info.config:
            return info.config[config_name]
        else:
            raise ValueError("Config section " + config_name + " not found")
    else:
        return info