p83571920
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

!1264 dataset: check whether input column list exist first in TFRecord file 

Merge pull request !1264 from ms_yan/tf_column
This commit is contained in:
mindspore-ci-bot 2020-05-19 21:50:20 +08:00 committed by Gitee
parent fe8b59f26b e6c25859d4
commit a4613b3107
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
mindspore/ccsrc/dataset/engine/datasetops/source/tf_reader_op.cc
View File

@ -918,6 +918,9 @@ Status TFReaderOp::CreateSchema(const std::string tf_file, const std::vector<std
[](const auto &it) -> std::string { return it.first; });
for (const auto &curr_col_name : columns) {
auto it = feature_map.find(curr_col_name);
if (it == feature_map.end()) {
RETURN_STATUS_UNEXPECTED("Failed to find column " + curr_col_name);
}
std::string column_name = it->first;
std::string column_type;