05281430

app/model/task_worker.py

@@ -212,7 +212,6 @@ class CDQTaskPre(object):
 
     def cdq_task(self, config):
         """场站级训练任务"""
-        station_id = -99
         try:
             print("111")
             # 动态生成场站数据路径
@@ -232,7 +231,7 @@ class CDQTaskPre(object):
             out_dir_cdq.replace('IN', 'OUT')
             weighted_cdq.to_csv(out_dir_cdq, index=False)
             print("555")
-            return {'status': 'success', 'station_id': station_id, 'weights': local_weights}
+            return {'cdq_status': 'success'}
         except Exception as e:
-            logging.error(f"Station {station_id} failed: {str(e)}")
-            return {'status': 'failed', 'station_id': station_id}
+            logging.error(f"cdq predict failed: {str(e)}")
+            return {'cdq_status': 'failed'}