Great job! Really cool solution here. I agree with the other comment that tqdm is simpler for iterables. Maybe see if you can abstract some of your logic to provide a wrapper function like tqdm does.