Hello, I found that when using specific snapshot times and not adding the end time of the simulation to the snapshot times the time integration gets stuck due to the time not increasing anymore because of this lines in time_integration:
if config.use_specific_snapshot_timepoints and config.return_snapshots: dt = jnp.minimum( dt, params.snapshot_timepoints[snapshot_data.current_checkpoint] - time )
As such when snapshot_data.current_checkpoint goes over the length of the snapshot_timepoints, the right part of the minimum calculation gives 0.0 all the time which gets selected over whatever dt is at that point in the time integration, therefore being stuck in an infinite loop.
Hello, I found that when using specific snapshot times and not adding the end time of the simulation to the snapshot times the time integration gets stuck due to the time not increasing anymore because of this lines in time_integration:
if config.use_specific_snapshot_timepoints and config.return_snapshots: dt = jnp.minimum( dt, params.snapshot_timepoints[snapshot_data.current_checkpoint] - time )As such when snapshot_data.current_checkpoint goes over the length of the snapshot_timepoints, the right part of the minimum calculation gives 0.0 all the time which gets selected over whatever dt is at that point in the time integration, therefore being stuck in an infinite loop.