checkpointing
try in py