init column_default_list_ when it is empty

pull/3369/head
jiangzhiwen 5 years ago
parent 859fe6bc41
commit 3bed1ef967

@ -758,6 +758,11 @@ Status CsvOp::ComputeColMap() {
} else { } else {
MS_LOG(WARNING) << "Column name map is already set!"; 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::Record<std::string>>(CsvOp::STRING, ""));
}
}
return Status::OK(); return Status::OK();
} }
} // namespace dataset } // namespace dataset

Loading…
Cancel
Save