Skip to content

Commit be86e34

Browse files
committed
Fix KeyError
1 parent 1377ba2 commit be86e34

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

preprocessing/me_binarizer.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -41,8 +41,8 @@ def __init__(self, config: dict):
4141
self.lr = LengthRegulator().to(self.device)
4242
self.merge_rest = self.binarization_args['merge_rest']
4343
self.merge_slur = self.binarization_args['merge_slur']
44-
self.slur_tolerance = self.binarization_args['slur_tolerance']
45-
self.round_midi = self.binarization_args['round_midi']
44+
self.slur_tolerance = self.binarization_args.get('slur_tolerance')
45+
self.round_midi = self.binarization_args.get('round_midi', False)
4646
self.key_shift_min, self.key_shift_max = self.config['key_shift_range']
4747

4848
def load_meta_data(self, raw_data_dir: pathlib.Path, ds_id):

0 commit comments

Comments
 (0)