diff --git a/ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/vo/ModelMetaVo.java b/ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/vo/ModelMetaVo.java
index af902cb3..f39e79e2 100644
--- a/ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/vo/ModelMetaVo.java
+++ b/ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/vo/ModelMetaVo.java
@@ -3,6 +3,7 @@ package com.ruoyi.platform.vo;
 import com.fasterxml.jackson.databind.PropertyNamingStrategy;
 import com.fasterxml.jackson.databind.annotation.JsonNaming;
 import com.ruoyi.platform.domain.dependencydomain.ProjectDepency;
+import com.ruoyi.platform.domain.dependencydomain.TrainTaskDepency;
 import io.swagger.annotations.ApiModelProperty;
 import lombok.Data;
 
@@ -53,7 +54,7 @@ public class ModelMetaVo implements Serializable {
     private HashMap<String, String> metrics;
 
     @ApiModelProperty(value = "训练任务")
-    private String trainTask;
+    private TrainTaskDepency trainTask;
 
     @ApiModelProperty(value = "模型标签（模型能力）")
     private String modelTag;