yhliang
2023-09-27 9366bd9bcfa093f33e8636e76e668cef6fc2b581
funasr/datasets/large_datasets/datapipes/filter.py
@@ -13,7 +13,7 @@
        self.fn = fn
    def set_epoch(self, epoch):
        self.epoch = epoch
        self.datapipe.set_epoch(epoch)
    def __iter__(self):
        assert callable(self.fn)
@@ -21,4 +21,4 @@
            if self.fn(data):
                yield data
            else:
                continue
                continue