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

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

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