Docs and code bug

[oiabtt]

in docs(https://miditok.readthedocs.io/en/latest/examples.html), you pass list

midi_paths = list(Path("path", "to", "dataset").glob("**/*.mid"))
augment_midi_dataset(
    midi_paths,

but in code, it's different

MidiTok/miditok/data_augmentation/data_augmentation.py

Line 17 in e6bdebd

data_path: Path | str,

MidiTok/miditok/data_augmentation/data_augmentation.py

Line 38 in e6bdebd

:param data_path: root path to the folder containing tokenized json files.

data_path: Path | str,

and ".midi" is ignored

MidiTok/miditok/data_augmentation/data_augmentation.py

Line 76 in e6bdebd

files_paths = list(Path(data_path).glob("**/*.mid"))

[Natooz]

Hi, thank you for the catch!
Indeed this should be fixed! For the file extensions we can use miditok.constants.MIDI_FILES_EXTENSIONS
Do you want to make the code contribution or should I do it?

[oiabtt]

Hi, thank you for the catch! Indeed this should be fixed! For the file extensions we can use miditok.constants.MIDI_FILES_EXTENSIONS Do you want to make the code contribution or should I do it?

you could do it