"""
Models in NAS aim to provide searched models from AutoML.
"""
from prototype.nas.models.utils.register import Registry

SPRING_MODELS_REGISTRY = Registry()

imported_vars = list(globals().items())

for var_name, var in imported_vars:
    if callable(var):
        SPRING_MODELS_REGISTRY.register(var_name, var)