From 94e916b0c23b1688f877ef13635b1585d4510a49 Mon Sep 17 00:00:00 2001
From: Mahdi <mahdi.rahmani.hanzaki@huawei.com>
Date: Fri, 23 Oct 2020 16:23:20 -0400
Subject: [PATCH] Fixed duplicate column problem in BucketBatchByLength

---
 .../engine/ir/datasetops/bucket_batch_by_length_node.cc       | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/mindspore/ccsrc/minddata/dataset/engine/ir/datasetops/bucket_batch_by_length_node.cc b/mindspore/ccsrc/minddata/dataset/engine/ir/datasetops/bucket_batch_by_length_node.cc
index 7f9d7f26ff..49bae0ac60 100644
--- a/mindspore/ccsrc/minddata/dataset/engine/ir/datasetops/bucket_batch_by_length_node.cc
+++ b/mindspore/ccsrc/minddata/dataset/engine/ir/datasetops/bucket_batch_by_length_node.cc
@@ -94,6 +94,10 @@ Status BucketBatchByLengthNode::ValidateParams() {
     }
   }
 
+  if (!column_names_.empty()) {
+    RETURN_IF_NOT_OK(ValidateDatasetColumnParam("BucketBatchByLengthNode", "column_names", column_names_));
+  }
+
   // Check bucket_batch_sizes: must be positive
   if (bucket_batch_sizes_.empty()) {
     std::string err_msg = "BucketBatchByLengthNode: bucket_batch_sizes must be non-empty";