Fix bug in dataframe

utils.py

@@ -144,14 +144,14 @@ def empty_note_sequence(qpm=120, total_time=0.0):
 
 # Generate piano_roll
 def sequence_to_pandas_dataframe(sequence):
-      pd_dict = collections.defaultdict(list)
-      for note in sequence.notes:
-          pd_dict["start_time"].append(note.start_time)
-          pd_dict["end_time"].append(note.end_time)
-          pd_dict["duration"].append(note.end_time - note.start_time)
-          pd_dict["pitch"].append(note.pitch)
-
-          return pd.DataFrame(pd_dict)
+    pd_dict = collections.defaultdict(list)
+    for note in sequence.notes:
+        pd_dict["start_time"].append(note.start_time)
+        pd_dict["end_time"].append(note.end_time)
+        pd_dict["duration"].append(note.end_time - note.start_time)
+        pd_dict["pitch"].append(note.pitch)
+        
+    return pd.DataFrame(pd_dict)
 
 def dataframe_to_pianoroll_img(df):
     fig = plt.figure(figsize=(8, 5))