File tree Expand file tree Collapse file tree 1 file changed +4
-4
lines changed
Expand file tree Collapse file tree 1 file changed +4
-4
lines changed Original file line number Diff line number Diff line change @@ -404,14 +404,14 @@ arrow::Status ArrowShuffleDictionaryWriter::serialize(arrow::io::OutputStream* o
404404 ARROW_RETURN_NOT_OK (out->Write (bitMap.data (), bitMapSize));
405405
406406 for (auto fieldIdx : dictionaryFields_) {
407+ auto it = dictionaries_.find (fieldIdx);
407408 GLUTEN_DCHECK (
408- dictionaries_. find (fieldIdx) != dictionaries_.end (),
409+ it != dictionaries_.end (),
409410 " Invalid dictionary field index: " + std::to_string (fieldIdx));
410411
411- const auto & dictionary = dictionaries_[fieldIdx];
412- ARROW_RETURN_NOT_OK (dictionary->serialize (out));
412+ ARROW_RETURN_NOT_OK (it->second ->serialize (out));
413413
414- dictionaries_.erase (fieldIdx );
414+ dictionaries_.erase (it );
415415 }
416416
417417 return arrow::Status::OK ();
You can’t perform that action at this time.
0 commit comments