import importlib def get_obj_from_str(string): module, cls = string.rsplit(".", 1) return getattr(importlib.import_module(module), cls) def build_from_config(config, **kwargs): if "target" not in config: raise KeyError("Expected key `target` to instantiate.") cls = get_obj_from_str(config["target"]) params = dict(config.get("params", {})) params.update(kwargs) pretrained_path = config.get("pretrained", None) if pretrained_path is not None and hasattr(cls, "from_pretrained"): return cls.from_pretrained(pretrained_path, **params) obj = cls(**params) return obj