提交 e6c25859 编写于 作者: M ms_yan

check wether input column exist in TFRecord files

上级 017cdbe8
......@@ -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;
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册