Update dataset.py (#375)

dbiir · Aug 24, 2023 · a68c3b3 · a68c3b3
1 parent ba8f95a
commit a68c3b3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/uer/utils/dataset.py b/uer/utils/dataset.py
@@ -553,7 +553,7 @@ def worker(self, proc_id, start, end):
                 if pos >= end:
                     break
 
-            dataset_writer.close()
+        dataset_writer.close()
 
 
 class T5Dataset(MlmDataset):