Custom Checkpointing example

The “session.get_checkpoint()” never get called in the following example on Custom Checkpointing:

Custom Checkpointing Example — Ray 2.1.0

I expect the following command to be called when running the script a second time. Any hints would be appreciated:

if session.get_checkpoint():
    loaded_checkpoint = session.get_checkpoint()
    step = loaded_checkpoint.to_dict()["step"] + 1