import os class EarlyLoopBreak(): def __init__(self, break_at: int): self.iter = 0 self.break_at = break_at self.on = bool(os.environ.get("EBREAK")) def on_break(self): if not self.on: return self.iter += 1 if self.break_at > 0 and self.iter >= self.break_at: return True