Exactly.

I am checkpointing the simulation manually, inside the step() method of my
own class that inherits from Simulation class.

What I am doing is checkpointing the simulation on every X steps (for
example, on every 5 steps, application saves the simulation). This automatic
checkpoint is important for me.

Do you have any suggestion on solving this? Maybe moving the checkpoint to
outside step() method, but is important to verify the current step, to
decide to save or not that moment.

Thank you very much.