from pathlib import Path
import variables as v
from datasets.primates_librosa_datamodule import PrimatesLibrosaDatamodule

data_root = Path() / 'data'


if __name__ == '__main__':
    dataset = PrimatesLibrosaDatamodule(data_root, batch_size=25, num_worker=6,
                                        sr=v.sr, n_mels=64, n_fft=512, hop_length=256)
    dataset.prepare_data()
    print('done')