LengthedGeneratorWrapper.py

from tqdm.auto import tqdm
import time

class LengthedGeneratorWrapper:
    """wraps an infinite generator with length for tqdm"""
    def __init__(self, infinite_generator, len):
        self.generator = infinite_generator
        self.len = len
    def __len__(self):
        return self.len
    def __iter__(self):
        for _ in range(self.len):
            yield next(self.generator)

def gen():
    while True:
        yield 0
g = gen()
dl = CustomSizeLoader(g, 1000)
for i in tqdm(dl):
    time.sleep(0.1)