models = {}


def register(name):
    def decorator(cls):
        models[name] = cls
        return cls
    return decorator


def make(name, config, dataset=None):
    model = models[name](config)
    return model

from . import gsmodel