Fix passing wrong type of parameters

omnizart/utils.py

@@ -11,6 +11,7 @@
 
 import jsonschema
 import pretty_midi
+import numpy as np
 import scipy.io.wavfile as wave
 
 from omnizart.constants.midi import SOUNDFONT_PATH
@@ -363,7 +364,7 @@ def aggregate_f0_info(pred, t_unit):
     start_idx = 0
     last_hz = pred[0]
     eps = 1e-6
-    pred.append(0)  # Append an additional zero to the end temporarily.
+    pred = np.append(pred, 0)  # Append an additional zero to the end temporarily.
     while cur_idx < len(pred):
         cur_hz = pred[cur_idx]
         if abs(cur_hz - last_hz) < eps:
@@ -390,5 +391,5 @@ def aggregate_f0_info(pred, t_unit):
         cur_idx += 1
         last_hz = cur_hz
 
-    del pred[-1]  # Remove the additional ending zero.
+    pred = pred[:-1]  # Remove the additional ending zero.
     return results

tests/test_utils.py

@@ -2,6 +2,7 @@
 import shutil
 
 import pytest
+import numpy as np
 from jsonschema import ValidationError
 
 from omnizart import utils
@@ -187,7 +188,7 @@ def test_serializable_recursive_value_path():
 
 def test_aggregate_f0_info():
     t_unit = 0.01
-    data = [0, 0, 0, 440, 440, 440, 440, 0, 0, 0, 220, 220]
+    data = np.array([0, 0, 0, 440, 440, 440, 440, 0, 0, 0, 220, 220])
     expected = [
         {"start_time": 0.03, "end_time": 0.07, "frequency": 440, "pitch": 69},
         {"start_time": 0.1, "end_time": 0.12, "frequency": 220, "pitch": 57}