游雁
2024-06-12 be26169447b2e5f8f38c97af8f5f6a201bc6ce40
funasr/datasets/large_datasets/datapipes/filter.py
@@ -1,14 +1,13 @@
from torch.utils.data import IterableDataset
def default_fn(data):
    return data
class FilterIterDataPipe(IterableDataset):
    def __init__(self,
                 datapipe,
                 fn=default_fn):
    def __init__(self, datapipe, fn=default_fn):
        self.datapipe = datapipe
        self.fn = fn