Use ArrowWriter as context manager in metric

aiinnova · Mar 2, 2021 · 7f287fb · 7f287fb
1 parent e00d72a
commit 7f287fb
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/datasets/metric.py b/src/datasets/metric.py
@@ -336,7 +336,8 @@ def _finalize(self):
         from all the nodes if main node or all_process is True.
         """
         if self.writer is not None:
-            self.writer.finalize()
+            with self.writer as writer:
+                writer.finalize()
         self.writer = None
         if self.filelock is not None:
             self.filelock.release()