fix

EleutherAI · Dec 6, 2024 · 00da281 · 00da281
1 parent 0a81439
commit 00da281
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/aria/datasets.py b/aria/datasets.py
@@ -448,7 +448,7 @@ def _get_mididicts_mp(_paths):
                         entry.metadata[k] = v
 
                 cnt += 1
-                # writer.write(entry.get_msg_dict())
+                writer.write(entry.get_msg_dict())
 
     logger.info(
         f"Finished - added {cnt}/{len(paths)} found MIDI files to dataset."