Wrong annotations for DenseInfos

[Randl]

DenseInfos are defined of shape times

diffrax/diffrax/_custom_types.py

Line 52 in 38a40e7

DenseInfos = dict[str, PyTree[Shaped[Array, "times ..."]]]

While in fact it probably should be times+1:

diffrax/diffrax/_global_interpolation.py

Line 322 in 38a40e7

assert _infos.shape[0] + 1 == self.ts.shape[0]

diffrax/diffrax/_integrate.py

Lines 883 to 887 in 38a40e7

	dense_ts = jnp.full(max_steps + 1, jnp.inf, dtype=time_dtype)
	_make_full = lambda x: jnp.full(
	(max_steps,) + jnp.shape(x), jnp.inf, dtype=x.dtype
	)
	dense_infos = jtu.tree_map(_make_full, dense_info)

As such global interpolation tests fail, e.g., see #391 but also locally on master

It is possibly related to some update, maybe of jaxtyping (this PR is on the new Lineax version).