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

"""
Forecasting 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 .forecasts import Forecast, PredictionBatch
from .predictions import ModelPerformanceMetric, PredictionCache
from .validation_run import ValidationRun
from .sales_data_update import SalesDataUpdate

# List all models for easier access
__all__ = [
    "Forecast",
    "PredictionBatch",
    "ModelPerformanceMetric",
    "PredictionCache",
    "ValidationRun",
    "SalesDataUpdate",
    "AuditLog",
]