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;