From f6d19e57b685bb15dabd3485ea35e15acd9d462f Mon Sep 17 00:00:00 2001 From: Urtzi Alfaro Date: Fri, 25 Jul 2025 20:01:37 +0200 Subject: [PATCH] Fix trainign model issue --- services/training/app/models/training.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/services/training/app/models/training.py b/services/training/app/models/training.py index 9cf5ead1..81f3ef11 100644 --- a/services/training/app/models/training.py +++ b/services/training/app/models/training.py @@ -5,12 +5,10 @@ Database models for training service from sqlalchemy import Column, Integer, String, DateTime, Text, Boolean, JSON, Float from sqlalchemy.dialects.postgresql import UUID, ARRAY -from sqlalchemy.ext.declarative import declarative_base +from shared.database.base import Base from datetime import datetime import uuid -Base = declarative_base() - class ModelTrainingLog(Base): """ Table to track training job execution and status.