From f62736ce09e7c16f51e2a00cc192dd4f91c90f25 Mon Sep 17 00:00:00 2001
From: cp3hnu <709011834@qq.com>
Date: Wed, 23 Oct 2024 14:49:28 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E6=94=B9=E5=AE=9E=E9=AA=8C=E5=AF=B9?=
 =?UTF-8?q?=E6=AF=94=E5=88=86=E9=A1=B5=E6=9F=A5=E8=AF=A2?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../platform/service/impl/AimServiceImpl.java  | 18 ++++++++++--------
 1 file changed, 10 insertions(+), 8 deletions(-)

diff --git a/ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/service/impl/AimServiceImpl.java b/ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/service/impl/AimServiceImpl.java
index a2e912e4..9eba931a 100644
--- a/ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/service/impl/AimServiceImpl.java
+++ b/ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/service/impl/AimServiceImpl.java
@@ -296,13 +296,15 @@ public class AimServiceImpl implements AimService {
     private List<String> getDataset(String isTrain, Map<String, Object> metricRecord) {
         List<String> datasetList = new ArrayList<>();
         List<Map<String, Object>> trainMetricRecords = (List<Map<String, Object>>) metricRecord.get(isTrain);
-        for (Map<String, Object> trainMetricRecord : trainMetricRecords) {
-            String taskId = (String) trainMetricRecord.get("task_id");
-            if (taskId.startsWith("model-" + isTrain)) {
-                List<Map<String, Object>> datasets = (List<Map<String, Object>>) trainMetricRecord.get("datasets");
-                for (Map<String, Object> dataset : datasets) {
-                    String datasetName = dataset.get("dataset_name") + ":" + dataset.get("dataset_version");
-                    datasetList.add(datasetName);
+        if (trainMetricRecords != null) {
+            for (Map<String, Object> trainMetricRecord : trainMetricRecords) {
+                String taskId = (String) trainMetricRecord.get("task_id");
+                if (taskId.startsWith("model-" + isTrain)) {
+                    List<Map<String, Object>> datasets = (List<Map<String, Object>>) trainMetricRecord.get("datasets");
+                    for (Map<String, Object> dataset : datasets) {
+                        String datasetName = dataset.get("dataset_name") + ":" + dataset.get("dataset_version");
+                        datasetList.add(datasetName);
+                    }
                 }
             }
         }
@@ -315,7 +317,7 @@ public class AimServiceImpl implements AimService {
         HashMap<String, Object> metrics = new HashMap<>();
         List<String> metricsNames = new ArrayList<>();
 
-        if(metricValues != null){
+        if (metricValues != null) {
             for (String key : metricValues.keySet()) {
                 Map<String, Object> valueMap = (Map<String, Object>) metricValues.get(key);
                 aimRunInfo.setRunId((String) valueMap.get("run_hash"));