added slakh to download.py

spotify · Jul 17, 2024 · becddb2 · becddb2
1 parent 76d5c9a
commit becddb2
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/basic_pitch/data/download.py b/basic_pitch/data/download.py
@@ -20,11 +20,12 @@
 from basic_pitch.data import commandline
 from basic_pitch.data.datasets.guitarset import main as guitarset_main
 from basic_pitch.data.datasets.ikala import main as ikala_main
+from basic_pitch.data.datasets.slakh import main as slakh_main
 
 logger = logging.getLogger()
 logger.setLevel(logging.INFO)
 
-DATASET_DICT = {"guitarset": guitarset_main, "ikala": ikala_main}
+DATASET_DICT = {"guitarset": guitarset_main, "ikala": ikala_main, "slakh": slakh_main}
 
 
 def main() -> None: