removed two pyright: ignore comments

patrick-kidger · May 9, 2024 · 7b9b1e7 · 7b9b1e7
1 parent 4f36bb3
commit 7b9b1e7
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git 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
@@ -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: