Fix evaluation

AIcrowd · May 7, 2021 · d370a87 · d370a87
1 parent 73dc139
commit d370a87
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/evaluator/music_demixing.py b/evaluator/music_demixing.py
@@ -91,7 +91,7 @@ def sdr(references, estimates):
             estimates = []
             for instrument in instruments:
                 reference_file = join(self.test_data_path, music_name, instrument + ".wav")
-                estimate_file = self.get_music_file_location(music_name, "bass")
+                estimate_file = self.get_music_file_location(music_name, instrument)
                 reference, _ = sf.read(reference_file)
                 estimate, _ = sf.read(estimate_file)
                 references.append(reference)