<filename>metric/src/models/__init__.py
from models.resnet import ResNet


def get_model(config, dset_config):
    if "resnet" in config.model.name:
        return ResNet(config, dset_config)
    else:
        raise NotImplementedError

<filename>metric/src/models/__init__.py
from models.resnet import ResNet


def get_model(config, dset_config):
    if "resnet" in config.model.name:
        return ResNet(config, dset_config)
    else:
        raise NotImplementedError
