def get_trainer_params()

in torchrecipes/utils/trainer_plugins.py [0:0]

• 11 lines of code
• 6 McCabe index (conditional complexity)

def get_trainer_params(trainer_conf: TrainerConf) -> Dict[str, Any]:
    if not isinstance(trainer_conf, DictConfig):
        # pyre-fixme[6]: Expected `str` for 1st param but got `TrainerConf`.
        trainer_conf = OmegaConf.create(trainer_conf)
    trainer_params = OmegaConf.to_container(trainer_conf, resolve=True) or {}
    assert isinstance(trainer_params, Dict)
    plugins = trainer_params.get("plugins", []) or []
    trainer_params["plugins"] = convert_trainer_plugins(
        trainer_params.get("precision", 32),
        plugins,
    )
    return {str(key): value for key, value in trainer_params.items()}