提交 50e5c7bf 编写于 作者: M mindspore-ci-bot 提交者: Gitee

!3369 Init CSV column default list when it's empty

Merge pull request !3369 from jiangzhiwen/fix_column_default
......@@ -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::Record<std::string>>(CsvOp::STRING, ""));
}
}
return Status::OK();
}
} // namespace dataset
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册