24 lines
477 B
Python
24 lines
477 B
Python
"""
|
|
Training 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 .training import (
|
|
TrainedModel,
|
|
ModelTrainingLog,
|
|
ModelPerformanceMetric,
|
|
TrainingJobQueue,
|
|
ModelArtifact,
|
|
)
|
|
|
|
# List all models for easier access
|
|
__all__ = [
|
|
"TrainedModel",
|
|
"ModelTrainingLog",
|
|
"ModelPerformanceMetric",
|
|
"TrainingJobQueue",
|
|
"ModelArtifact",
|
|
]
|