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

Merge pull request !6120 from jiangzhiwen/fix/prob_nullptr
4 years ago · 1a5d4b8ed1
parent 3ed015e645 2a59596fe6
commit 1a5d4b8ed1
1 changed files with 1 additions and 1 deletions
--- a/mindspore/ccsrc/minddata/dataset/api/iterator.cc
+++ b/mindspore/ccsrc/minddata/dataset/api/iterator.cc
@ -123,9 +123,9 @@ Status Iterator::BuildAndLaunchTree(std::shared_ptr<Dataset> ds) {

  // Launch the execution tree.
  RETURN_IF_NOT_OK(tree_->Prepare());
+  tree_->Launch();
  iterator_ = std::make_unique<DatasetIterator>(tree_);
  RETURN_UNEXPECTED_IF_NULL(iterator_);
-  RETURN_IF_NOT_OK(tree_->Launch());

  return rc;
 }