src/model/__init__.py

from src.model.styleRF import StyleRF
from src.utils.registry import Registry

MODEL_REGISTRY = Registry("MODEL")

MODEL_REGISTRY.register(StyleRF)