Fixing egfxset expected return value

mir-dataset-loaders · Mar 1, 2023 · c71de62 · c71de62
1 parent 64a3307
commit c71de62
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/mirdata/datasets/egfxset.py b/mirdata/datasets/egfxset.py
@@ -205,7 +205,7 @@ class Track(core.Track):
         setting (list): the setting of the effect recorded or "None" when the recording is a clean effect sound
 
     Cached Properties:
-        note_name (list): a list with the note name annotation of the audio file (e.g. "Ab5", "C6", etc.)
+        note_name (ndarray): a list with the note name annotation of the audio file (e.g. "Ab5", "C6", etc.)
         midinote (NoteData): the midinote annotation of the audio file
     """
 

diff --git a/tests/datasets/test_egfxset.py b/tests/datasets/test_egfxset.py
@@ -1,4 +1,5 @@
 import os
+import numpy as np
 from mirdata import annotations
 from mirdata.datasets import egfxset
 from tests.test_utils import run_track_tests
@@ -35,7 +36,7 @@ def test_track():
 
     expected_property_types = {
         "audio": tuple,
-        "note_name": list,
+        "note_name": np.ndarray,
         "midinote": annotations.NoteData,
     }
 
@@ -74,7 +75,7 @@ def test_track():
 
     expected_property_types = {
         "audio": tuple,
-        "note_name": list,
+        "note_name": np.ndarray,
         "midinote": annotations.NoteData,
     }