"""mopet: parameter-efficient mixture-of-experts for unified medical image classification."""

from __future__ import annotations

from ._factory import create_model, list_pretrained, load_pretrained_weights
from ._moe import MoEModule, apply_moe_peft
from .model import MoPET, MultiTaskClassifier

__version__ = "0.1.2"
__all__ = [
    "MoPET",
    "MultiTaskClassifier",
    "MoEModule",
    "apply_moe_peft",
    "create_model",
    "list_pretrained",
    "load_pretrained_weights",
    "__version__",
]