fix ut

apache · Jul 30, 2024 · e60c27e · e60c27e
1 parent 96a2afd
commit e60c27e
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/cpp/core/shuffle/LocalPartitionWriter.cc b/cpp/core/shuffle/LocalPartitionWriter.cc
@@ -490,10 +490,9 @@ arrow::Status LocalPartitionWriter::stop(ShuffleWriterMetrics* metrics) {
       ARROW_ASSIGN_OR_RAISE(endInFinalFile, dataFileOs_->Tell());
       partitionLengths_[pid] = endInFinalFile - startInFinalFile;
     }
-    // Close Final file. Clear buffered resources.
-    RETURN_NOT_OK(clearResource());
   }
-
+  // Close Final file. Clear buffered resources.
+  RETURN_NOT_OK(clearResource());
   // Check all spills are merged.
   auto s = 0;
   for (const auto& spill : spills_) {