src/main/java/com/kamco/cd/training/model/service/ModelTrainDetailService.java

@@ -81,6 +81,7 @@ public class ModelTrainDetailService {
       datasetIds.add(mappingDataset.getDatasetId());
     }
     datasetReq.setIds(datasetIds);
+    datasetReq.setModelNo(modelInfo.getModelNo());
 
     if (modelInfo.getModelNo().equals("G1")) {
       dataSets = mngCoreService.getDatasetSelectG1List(datasetReq);