Browse Source

Fixed duplicate column problem in BucketBatchByLength

tags/v1.1.0
Mahdi 5 years ago
parent
commit
94e916b0c2
1 changed files with 4 additions and 0 deletions
  1. +4
    -0
      mindspore/ccsrc/minddata/dataset/engine/ir/datasetops/bucket_batch_by_length_node.cc

+ 4
- 0
mindspore/ccsrc/minddata/dataset/engine/ir/datasetops/bucket_batch_by_length_node.cc View File

@@ -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 // Check bucket_batch_sizes: must be positive
if (bucket_batch_sizes_.empty()) { if (bucket_batch_sizes_.empty()) {
std::string err_msg = "BucketBatchByLengthNode: bucket_batch_sizes must be non-empty"; std::string err_msg = "BucketBatchByLengthNode: bucket_batch_sizes must be non-empty";


Loading…
Cancel
Save