drop-in replacmement for time.sleep (python)

util.py

def wait(s, delay=1):
    """
    drop-in for time.sleep() with status updates and Ctrl-C support

    Usage:
        for delay in wait(seconds):
            print(f'waiting for {delay} seconds')
    """
    import time

    try:
        while s:
            yield str(s)
            time.sleep(delay)
            s -= 1
    except KeyboardInterrupt:
        pass

I hand-crafted this, then I got curious if ChatGPT could do it, here's how that went