yhliang
2023-09-27 945dae5f13c4ea37aa2d6b6e6f57eae1a4249b64
funasr/datasets/large_datasets/datapipes/map.py
@@ -14,7 +14,7 @@
        self.fn = fn
    def set_epoch(self, epoch):
        self.epoch = epoch
        self.datapipe.set_epoch(epoch)
    def __iter__(self):
        assert callable(self.fn)