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"));