checkpointing.refactor.util 

 1from itertools import count
 2
 3
 4def local_variable_names_generator():
 5    for i in count():
 6        yield f"__checkpointing_local_var_{i}__"
 7
 8def nonlocal_variable_names_generator():
 9    for i in count():
10        yield f"__checkpointing_nonlocal_var_{i}__"
def local_variable_names_generator() 
5def local_variable_names_generator():
6    for i in count():
7        yield f"__checkpointing_local_var_{i}__"
def nonlocal_variable_names_generator() 
 9def nonlocal_variable_names_generator():
10    for i in count():
11        yield f"__checkpointing_nonlocal_var_{i}__"