bakery-ia/services/training/app/models/__init__.py

"""
Training Service Models Package

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

# Import AuditLog model for this service
from shared.security import create_audit_log_model
from shared.database.base import Base

# Create audit log model for this service
AuditLog = create_audit_log_model(Base)

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

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