Implement arbitrary diagnostics functions for crosscat.state.State.transition

[fsaad]

Diagnostic func should take the state and performs arbitrary operations. Makes it easier to run transitions with arbitrary diagnostics at some amount of checkpoints, without the client having to pause inference, run diagnostics, and continue.