Fix new Frontend 13

services/training/app/services/training_service.py

@@ -28,6 +28,8 @@ from app.services.messaging import (
     publish_job_failed
 )
 
+from app.services.messaging import TrainingStatusPublisher
+
 logger = logging.getLogger(__name__)
 
 class TrainingService:
@@ -73,6 +75,7 @@ class TrainingService:
         
         logger.info(f"Starting training job {job_id} for tenant {tenant_id}")
         
+        self.status_publisher = TrainingStatusPublisher(job_id, tenant_id)
         
         try:
             
@@ -85,7 +88,12 @@ class TrainingService:
                 requested_end=requested_end,
                 job_id=job_id
             )
-            await publish_job_progress(job_id, tenant_id, 10, "data_validation", estimated_time_remaining_minutes=8)
+            
+            await self.status_publisher.progress_update(
+                progress=10,
+                step="data_validation",
+                step_details="Data validation and alignment completed"
+            )
             
             # Step 2: Execute ML training pipeline
             logger.info("Step 2: Starting ML training pipeline")