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

Fix bug in checking status return in CSVOp

This commit is contained in:
hesham 2021-04-20 19:20:22 -04:00
parent d35939603a
commit d167095b6d
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
mindspore/ccsrc/minddata/dataset/engine/datasetops/source/csv_op.cc
View File

@ -214,6 +214,7 @@ int CsvOp::CsvParser::PutRow(int c) {
Status s = rows_connector_->Add(worker_id_, std::move(cur_row_));
if (s.IsError()) {
err_message_ = s.ToString();
if (s.StatusCode() == kMDInterrupted) return -2;
return -1;
}
@ -502,7 +503,9 @@ Status CsvOp::LoadFile(const std::string &file, int64_t start_offset, int64_t en
// which is a 32-bit -1, it's not equal to the 8-bit -1 on Euler OS. So instead of char, we use
// int to receive its return value.
int chr = ifs.get();
if (csv_parser.ProcessMessage(chr) != 0) {
int err = csv_parser.ProcessMessage(chr);
if (err != 0) {
if (err == -2) return Status(kMDInterrupted);
RETURN_STATUS_UNEXPECTED("Invalid file, failed to parse file: " + file + ":" +
std::to_string(csv_parser.GetTotalRows() + 1) +
". Error message: " + csv_parser.GetErrorMessage());