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)