diff --git a/diffrax/_progress_meter.py b/diffrax/_progress_meter.py index 9b8d0a1b..8a813be6 100644 --- a/diffrax/_progress_meter.py +++ b/diffrax/_progress_meter.py @@ -294,7 +294,7 @@ def _step(_progress, _idx): # Return the idx to thread the callbacks in the correct order. return _idx - return jax.pure_callback(_step, idx, progress, idx, vectorized=True) # pyright: ignore + return jax.pure_callback(_step, idx, progress, idx, vectorized=True) def close(self, close_bar: Callable[[Any], None], idx: IntScalarLike): def _close(_idx): diff --git a/test/helpers.py b/test/helpers.py index b7e33b54..66970db7 100644 --- a/test/helpers.py +++ b/test/helpers.py @@ -159,7 +159,7 @@ def _batch_sde_solve( shape=struct, tol=bm_tol, key=key, - levy_area=concrete_la, # pyright: ignore + levy_area=concrete_la, ) terms = get_terms(bm) if controller is None: