salu133445 / muspy

A toolkit for symbolic music generation
https://salu133445.github.io/muspy/
MIT License
427 stars 49 forks source link

No compression support in MusicDataset #59

Closed cifkao closed 3 years ago

cifkao commented 3 years ago

MusicDataset doesn't seem to support loading compressed files (#39). Neither kind="json" nor kind="json.gz" works.

cifkao commented 3 years ago

Here is a workaround:

class MusicDataset(muspy.datasets.MusicDataset):

    def __init__(self, root: Union[str, Path], kind: str = "json", suffix: str = "json"):
        super().__init__(root=root, kind=kind)
        self.filenames = sorted(self.root.rglob("*." + suffix))

dataset = MusicDataset('data/', kind='json', suffix='json.gz')