Files
bakery-ia/services/forecasting/app/models/__init__.py
2025-11-18 07:17:17 +01:00

29 lines
794 B
Python

"""
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",
]