Update basic_pitch/constants.py

Co-authored-by: David Rubinstein <[email protected]>
spotify · Aug 16, 2024 · cebd1d6 · cebd1d6
1 parent 1a2dfa6
commit cebd1d6
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/basic_pitch/constants.py b/basic_pitch/constants.py
@@ -62,4 +62,7 @@ def _freq_bins(bins_per_semitone: int, base_frequency: float, n_semitones: int)
 FREQ_BINS_NOTES = _freq_bins(NOTES_BINS_PER_SEMITONE, ANNOTATIONS_BASE_FREQUENCY, ANNOTATIONS_N_SEMITONES)
 FREQ_BINS_CONTOURS = _freq_bins(CONTOURS_BINS_PER_SEMITONE, ANNOTATIONS_BASE_FREQUENCY, ANNOTATIONS_N_SEMITONES)
 
-Split = Enum("Split", ["train", "validation", "test"])
+class Split(Enum):
+    train = "train"
+    validation = "validation"
+    test = "test"