Fix DB issues

2025-09-30 13:32:51 +02:00
parent ec6bcb4c7d
commit 147893015e
51 changed files with 341 additions and 1032 deletions
--- a/services/training/app/models/init.py
+++ b/services/training/app/models/init.py
@@ -0,0 +1,23 @@
+"""
+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",
+]