Fix checkpointing again

[hamishivi]

I removed a conversion to an int for the checkpointing_steps arg before, but turns out we needed it, im dumb.