Added genre to ballroom datataset loader

docs/source/table.rst

@@ -30,6 +30,7 @@
        - annotations: ✅
      - - :ref:`beats`
        - :ref:`tempo`
+       - :ref:`genre`
      - 698
      - .. image:: https://licensebuttons.net/l/zero/1.0/80x15.png
           :target: http://creativecommons.org/publicdomain/zero/1.0/

mirdata/datasets/ballroom.py

@@ -101,6 +101,7 @@ class Track(core.Track):
         audio_path (str): path to audio file
         beats_path (str): path to beats file
         tempo_path (str): path to tempo file
+        genre (str): genre of the track
 
     Cached Properties:
         beats (BeatData): human-labeled beat annotations
@@ -130,6 +131,8 @@ def __init__(
         self.beats_path = self.get_path("beats")
         self.tempo_path = self.get_path("tempo")
 
+        self.genre = os.path.basename(os.path.dirname(self.audio_path)).lower()
+
     @core.cached_property
     def beats(self) -> Optional[annotations.BeatData]:
         return load_beats(self.beats_path)

tests/datasets/test_ballroom.py

@@ -25,12 +25,14 @@ def test_track():
             os.path.normpath("tests/resources/mir_datasets/ballroom/"),
             "B_1.0/annotations/tempo/Media-105901.bpm",
         ),
+        "genre": "waltz",
     }
 
     expected_property_types = {
         "tempo": float,
         "beats": annotations.BeatData,
         "audio": tuple,
+        "genre": str,
     }
 
     run_track_tests(track, expected_attributes, expected_property_types)

tests/test_loaders.py

@@ -13,6 +13,7 @@
 DATASETS = mirdata.DATASETS
 CUSTOM_TEST_TRACKS = {
     "beatles": "0111",
+    "ballroom": "Media-105901",
     "cante100": "008",
     "compmusic_carnatic_rhythm": "10003",
     "compmusic_carnatic_varnam": "dharini_abhogi",