32 lines
707 B
Python
32 lines
707 B
Python
"""
|
|
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,
|
|
)
|
|
|
|
# List all models for easier access
|
|
__all__ = [
|
|
"TrainedModel",
|
|
"ModelTrainingLog",
|
|
"ModelPerformanceMetric",
|
|
"TrainingJobQueue",
|
|
"ModelArtifact",
|
|
"AuditLog",
|
|
]
|