src/main/java/com/kamco/cd/training/train/service/TrainJobWorker.java

@@ -22,6 +22,8 @@ public class TrainJobWorker {
 
   private final ModelTrainJobCoreService modelTrainJobCoreService;
   private final ModelTrainMngCoreService modelTrainMngCoreService;
+  private final ModelTrainMetricsJobService modelTrainMetricsJobService;
+  private final ModelTestMetricsJobService modelTestMetricsJobService;
   private final DockerTrainService dockerTrainService;
   private final ObjectMapper objectMapper;
 
@@ -90,8 +92,10 @@ public class TrainJobWorker {
 
         if (isEval) {
           modelTrainMngCoreService.markStep2Success(modelId);
+          modelTrainMetricsJobService.findTrainValidMetricCsvFiles();
         } else {
           modelTrainMngCoreService.markStep1Success(modelId);
+          modelTestMetricsJobService.findTestValidMetricCsvFiles();
         }
 
       } else {