Fix services

This commit is contained in:
Urtzi Alfaro
2025-07-17 19:24:57 +02:00
parent 599e335fbb
commit ea5b75b685
7 changed files with 142 additions and 4 deletions

View File

@@ -1,5 +1,5 @@
"""
Training models
Training models - Fixed version
"""
from sqlalchemy import Column, String, Integer, DateTime, Text, JSON, Boolean, Float
@@ -79,7 +79,7 @@ class TrainedModel(Base):
return f"<TrainedModel(id={self.id}, product={self.product_name}, tenant={self.tenant_id})>"
class TrainingLog(Base):
"""Training log entries"""
"""Training log entries - FIXED: renamed metadata to log_metadata"""
__tablename__ = "training_logs"
id = Column(UUID(as_uuid=True), primary_key=True, default=uuid.uuid4)
@@ -93,7 +93,7 @@ class TrainingLog(Base):
# Additional data
execution_time = Column(Float) # Time taken for this step
memory_usage = Column(Float) # Memory usage in MB
metadata = Column(JSON) # Additional metadata
log_metadata = Column(JSON) # FIXED: renamed from 'metadata' to 'log_metadata'
created_at = Column(DateTime, default=datetime.utcnow)