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

!6120 Fix c api of ImageFolderDataset probably fails in unit testing 

Merge pull request !6120 from jiangzhiwen/fix/prob_nullptr
This commit is contained in:
mindspore-ci-bot 2020-09-12 19:00:47 +08:00 committed by Gitee
parent 3ed015e645 2a59596fe6
commit 1a5d4b8ed1
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mindspore/ccsrc/minddata/dataset/api/iterator.cc
View File

@ -123,9 +123,9 @@ Status Iterator::BuildAndLaunchTree(std::shared_ptr<Dataset> ds) {
// Launch the execution tree. // Launch the execution tree.
RETURN_IF_NOT_OK(tree_->Prepare()); RETURN_IF_NOT_OK(tree_->Prepare());
tree_->Launch();
iterator_ = std::make_unique<DatasetIterator>(tree_); iterator_ = std::make_unique<DatasetIterator>(tree_);
RETURN_UNEXPECTED_IF_NULL(iterator_); RETURN_UNEXPECTED_IF_NULL(iterator_);
RETURN_IF_NOT_OK(tree_->Launch());
return rc; return rc;
} }