imporve features

services/training/app/api/models.py

@@ -11,7 +11,7 @@ from sqlalchemy import text
 from app.core.database import get_db
 from app.schemas.training import TrainedModelResponse, ModelMetricsResponse
 from app.services.training_service import EnhancedTrainingService
-from datetime import datetime
+from datetime import datetime, timezone
 from sqlalchemy import select, delete, func
 import uuid
 import shutil
@@ -79,13 +79,13 @@ async def get_active_model(
         
         # ✅ FIX: Wrap update query with text() too
         update_query = text("""
-            UPDATE trained_models 
-            SET last_used_at = :now 
+            UPDATE trained_models
+            SET last_used_at = :now
             WHERE id = :model_id
         """)
-        
+
         await db.execute(update_query, {
-            "now": datetime.utcnow(),
+            "now": datetime.now(timezone.utc),
             "model_id": model_record.id
         })
         await db.commit()
@@ -300,7 +300,7 @@ async def delete_tenant_models_complete(
         
         deletion_stats = {
             "tenant_id": tenant_id,
-            "deleted_at": datetime.utcnow().isoformat(),
+            "deleted_at": datetime.now(timezone.utc).isoformat(),
             "jobs_cancelled": 0,
             "models_deleted": 0,
             "artifacts_deleted": 0,
@@ -322,7 +322,7 @@ async def delete_tenant_models_complete(
             
             for job in active_jobs:
                 job.status = "cancelled"
-                job.updated_at = datetime.utcnow()
+                job.updated_at = datetime.now(timezone.utc)
                 deletion_stats["jobs_cancelled"] += 1
             
             if active_jobs: