From 3bed1ef9673188d2bd418fcd6f0b1ed129b1f67e Mon Sep 17 00:00:00 2001 From: jiangzhiwen Date: Thu, 23 Jul 2020 16:59:04 +0800 Subject: [PATCH] init column_default_list_ when it is empty --- .../minddata/dataset/engine/datasetops/source/csv_op.cc | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/mindspore/ccsrc/minddata/dataset/engine/datasetops/source/csv_op.cc b/mindspore/ccsrc/minddata/dataset/engine/datasetops/source/csv_op.cc index c7f36e9d760..0871b3f30cd 100644 --- a/mindspore/ccsrc/minddata/dataset/engine/datasetops/source/csv_op.cc +++ b/mindspore/ccsrc/minddata/dataset/engine/datasetops/source/csv_op.cc @@ -758,6 +758,11 @@ Status CsvOp::ComputeColMap() { } else { MS_LOG(WARNING) << "Column name map is already set!"; } + if (column_default_list_.size() < column_name_id_map_.size()) { + for (int32_t i = column_default_list_.size(); i < column_name_id_map_.size(); i++) { + column_default_list_.push_back(std::make_shared>(CsvOp::STRING, "")); + } + } return Status::OK(); } } // namespace dataset