Class DummyDataset

flowmap/dataset/data_module_overfit.py:7–17

class DummyDataset(IterableDataset):
    def __init__(self, limit: int | None = None) -> None:
        self.limit = limit

    def __iter__(self) -> Iterator:
        if self.limit:
            for _ in range(self.limit):
                yield []
        else:
            while True:
                yield []


class DataModuleOverfit(LightningDataModule):