diff --git a/wekws/dataset/dataset.py b/wekws/dataset/dataset.py index cfcd00d..74aaf4d 100644 --- a/wekws/dataset/dataset.py +++ b/wekws/dataset/dataset.py @@ -113,7 +113,7 @@ class DataList(IterableDataset): yield data -def Dataset(data_list_file, conf, +def Dataset(data_list_file, conf, partition=True, reverb_lmdb=None, noise_lmdb=None):