AstrAI/astrai/config/__init__.py

from astrai.config.model_config import ModelConfig
from astrai.config.train_config import TrainConfig

__all__ = [
    # Model configuration
    "ModelConfig",
    "TrainConfig",
]