Make sure the context variable is defined properly. Also, try removing checkpoint= and just pass checkpoint directly as a positional argument (maybe the function add_or_update_checkpoint does not expect a keyword argument).
Hello @jakeshivers
Your answer to this exercise is correct. I tried several things to reproduce the problem with no avail. However, I guess the problem should be with the checkpoint variable. Did you change anything in the cell in part 7? Or is it possible maybe you skipped this cell and forgot to run it?