"""
Training Service Models Package

Import all models to ensure they are registered with SQLAlchemy Base.
"""

# Import all models to register them with the Base metadata
from .training import (
    TrainedModel,
    ModelTrainingLog,
    ModelPerformanceMetric,
    TrainingJobQueue,
    ModelArtifact,
)

# List all models for easier access
__all__ = [
    "TrainedModel",
    "ModelTrainingLog",
    "ModelPerformanceMetric",
    "TrainingJobQueue",
    "ModelArtifact",
]