Fix resolve pedal (#106)

EleutherAI · Mar 10, 2024 · 791b2f0 · 791b2f0
1 parent 7c2c47e
commit 791b2f0
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/aria/data/midi.py b/aria/data/midi.py
@@ -247,7 +247,7 @@ def _resolve_overlaps(self):
         return self
 
     def resolve_pedal(self):
-        """Remove pedal_msgs by extending note offsets and resolving overlaps"""
+        """Resolve pedal - extend note offsets and resolve note overlaps"""
         # Organize note messages by channel
         note_msgs_c = defaultdict(list)
         for msg in self.note_msgs:
@@ -265,7 +265,6 @@ def resolve_pedal(self):
                         msg["data"]["end"] = pedal_end
                         break
 
-        self.pedal_msgs = []
         return self._resolve_overlaps()
 
     # TODO: