diff --git a/ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/service/impl/ExperimentServiceImpl.java b/ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/service/impl/ExperimentServiceImpl.java index 1a8fc568..49a233c2 100644 --- a/ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/service/impl/ExperimentServiceImpl.java +++ b/ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/service/impl/ExperimentServiceImpl.java @@ -581,7 +581,7 @@ public class ExperimentServiceImpl implements ExperimentService { if (trainDatasetList != null) { List trainDatasets = new ArrayList<>(); for (Map dataset : trainDatasetList) { - if(!dataset.containsKey("dataset_ref")) { + if (!dataset.containsKey("dataset_ref")) { NewDatasetVo newDatasetVo = new NewDatasetVo(); newDatasetVo.setId(Integer.valueOf((String) dataset.get("dataset_id"))); newDatasetVo.setName((String) dataset.get("dataset_name")); @@ -631,13 +631,15 @@ public class ExperimentServiceImpl implements ExperimentService { List> testDatasetList = (List>) evaluateMap.get("datasets"); List testDatasets = new ArrayList<>(); for (Map dataset : testDatasetList) { - NewDatasetVo newDatasetVo = new NewDatasetVo(); - newDatasetVo.setId(Integer.valueOf((String) dataset.get("dataset_id"))); - newDatasetVo.setName((String) dataset.get("dataset_name")); - newDatasetVo.setVersion((String) dataset.get("dataset_version")); - newDatasetVo.setIdentifier((String) dataset.get("dataset_identifier")); - newDatasetVo.setOwner((String) dataset.get("owner")); - testDatasets.add(newDatasetVo); + if (!dataset.containsKey("dataset_ref")) { + NewDatasetVo newDatasetVo = new NewDatasetVo(); + newDatasetVo.setId(Integer.valueOf((String) dataset.get("dataset_id"))); + newDatasetVo.setName((String) dataset.get("dataset_name")); + newDatasetVo.setVersion((String) dataset.get("dataset_version")); + newDatasetVo.setIdentifier((String) dataset.get("dataset_identifier")); + newDatasetVo.setOwner((String) dataset.get("owner")); + testDatasets.add(newDatasetVo); + } } modelMetaVo.setTestDatasets(testDatasets); }