Fix services
This commit is contained in:
@@ -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)
|
||||
|
||||
|
||||
Reference in New Issue
Block a user