From 7026dc32a514683ecd71a011f439a324747bf584 Mon Sep 17 00:00:00 2001
From: gaoxinxing <15031259256@163.com>
Date: Mon, 13 Nov 2023 17:42:38 +0800
Subject: [PATCH] add dataset info for step spec

---
 .../java/ai/starwhale/mlops/domain/job/spec/StepSpec.java   | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/server/controller/src/main/java/ai/starwhale/mlops/domain/job/spec/StepSpec.java b/server/controller/src/main/java/ai/starwhale/mlops/domain/job/spec/StepSpec.java
index 19a4ad2f04..ee140d21fc 100644
--- a/server/controller/src/main/java/ai/starwhale/mlops/domain/job/spec/StepSpec.java
+++ b/server/controller/src/main/java/ai/starwhale/mlops/domain/job/spec/StepSpec.java
@@ -82,6 +82,12 @@ public class StepSpec {
     @JsonProperty("require_dataset")
     private Boolean requireDataset;
 
+    @JsonProperty("require_train_datasets")
+    private Boolean requireTrainDataset;
+
+    @JsonProperty("require_validation_datasets")
+    private Boolean requireValidationDataset;
+
     @JsonProperty("container_spec")
     ContainerSpec containerSpec;