tools/pymss/modules/__init__.py

from ._core_shims import alias_submodules

alias_submodules(
    __name__,
    "pymss_core.modules",
    (
        "apollo_mlx",
        "bandit_mlx",
        "demucs4ht",
        "demucs_local",
        "demucs_mlx",
        "legacy_demucs",
        "mdx23c_mlx",
        "mdx23c_tfc_tdf_v3",
        "mlx_utils",
        "scnet_mlx",
        "spectrogram",
    ),
)

alias_submodules(
    __name__,
    "pymss_core.modules",
    (
        "bandit",
        "bandit.bandsplit",
        "bandit.core",
        "bandit.core.model",
        "bandit.core.model._spectral",
        "bandit.core.model.bsrnn",
        "bandit.core.model.bsrnn.bandsplit",
        "bandit.core.model.bsrnn.core",
        "bandit.core.model.bsrnn.maskestim",
        "bandit.core.model.bsrnn.tfmodel",
        "bandit.core.model.bsrnn.utils",
        "bandit.core.model.bsrnn.wrapper",
        "bandit.maskestim",
        "bandit.tfmodel",
        "bandit_v2",
        "bandit_v2.bandit",
        "bandit_v2.bandsplit",
        "bandit_v2.maskestim",
        "bandit_v2.tfmodel",
        "bandit_v2.utils",
        "bs_roformer",
        "bs_roformer.attend",
        "bs_roformer.bands",
        "bs_roformer.bs_roformer",
        "bs_roformer.bs_roformer_hyperace",
        "bs_roformer.common",
        "bs_roformer.hyperace_segm",
        "bs_roformer.mel_band_roformer",
        "bs_roformer.mlx_attention",
        "bs_roformer.mlx_roformer",
        "bs_roformer.transformer",
        "look2hear",
        "look2hear.apollo",
        "scnet",
        "scnet.scnet",
        "scnet.separation",
    ),
)