From b860e9a4d4c92525370a8d6d2574cbf5bdfa339a Mon Sep 17 00:00:00 2001 From: chenzhihang <709011834@qq.com> Date: Mon, 13 Jan 2025 17:03:58 +0800 Subject: [PATCH] =?UTF-8?q?=E4=BC=98=E5=8C=96=E6=B5=81=E6=B0=B4=E7=BA=BF?= =?UTF-8?q?=E8=BD=AC=E6=8D=A2bug?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../service/impl/ExperimentServiceImpl.java | 18 ++++++++++-------- 1 file changed, 10 insertions(+), 8 deletions(-) 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); }