Fix trainign model issue

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.