Improved sanity checks.
Improved sanity checks.

Catch the IntegratorMoveError to save the context causing errors.
Check only for NaNs in energy since checking NaNs in positions is redundant, and check also unsampled states energies.

choderalab/yank,choderalab/yank