Format checkpoint.py