src/model/encoder/__init__.py

from .encoder import Encoder
from .encoder_codecsplat import EncoderCodecSplat, EncoderCodecSplatCfg

ENCODERS = {
    "codecsplat": EncoderCodecSplat,
}

EncoderCfg = EncoderCodecSplatCfg


def get_encoder(cfg: EncoderCfg) -> Encoder:
    return ENCODERS[cfg.name](cfg)