From 41facc168ca19e9fee53623a02d94e286d0437cd Mon Sep 17 00:00:00 2001
From: Zirui Wu <zirui.wu@huawei.com>
Date: Tue, 27 Oct 2020 16:57:44 -0400
Subject: [PATCH] validate param added during tree prepare

---
 mindspore/ccsrc/minddata/dataset/engine/tree_adapter.cc | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/mindspore/ccsrc/minddata/dataset/engine/tree_adapter.cc b/mindspore/ccsrc/minddata/dataset/engine/tree_adapter.cc
index 37d2daaf8c..9b24a6344e 100644
--- a/mindspore/ccsrc/minddata/dataset/engine/tree_adapter.cc
+++ b/mindspore/ccsrc/minddata/dataset/engine/tree_adapter.cc
@@ -66,6 +66,8 @@ Status TreeAdapter::GetNext(TensorRow *row) {
 }
 
 Status TreeAdapter::DFSBuildTree(std::shared_ptr<api::Dataset> ir, std::shared_ptr<DatasetOp> *op) {
+  // validate the op can be built first before building the DatasetOp
+  RETURN_IF_NOT_OK(ir->ValidateParams());
   std::vector<std::shared_ptr<DatasetOp>> ops = ir->Build();
   CHECK_FAIL_RETURN_UNEXPECTED(!ops.empty(), "Unable to build node.");