ignore

Author: lockwo

diffrax_extensions/_misc.py

@@ -151,7 +151,7 @@ def static_select(pred: BoolScalarLike, a: ArrayLike, b: ArrayLike) -> ArrayLike
     # This in turn allows us to perform some trace-time optimisations that XLA isn't
     # smart enough to do on its own.
     if isinstance(pred, (np.ndarray, np.generic)) and pred.shape == ():
-        pred = pred.item()
+        pred = pred.item()  # pyright: ignore
     if pred is True:
         return a
     elif pred is False:

diffrax_extensions/_progress_meter.py

@@ -123,7 +123,7 @@ def _step_bar(bar: list[float], progress: FloatScalarLike) -> None:
         if eqx.is_array(progress):
             # May not be an array when called with `JAX_DISABLE_JIT=1`
             progress = cast(Union[Array, np.ndarray], progress)
-            progress = progress.item()
+            progress = progress.item()  # pyright: ignore
         progress = cast(float, progress)
         bar[0] = progress
         print(f"{100 * progress:.2f}%")

docs/api/terms.md

@@ -151,14 +151,14 @@ the friction and the damping of the system.
 These terms enable the use of ULD-specific solvers which can be found 
 [here](./solvers/sde_solvers.md#underdamped-langevin-solvers). Note that these ULD solvers will only work if given
 terms with structure `MultiTerm(UnderdampedLangevinDriftTerm(gamma, u, grad_f), UnderdampedLangevinDiffusionTerm(gamma, u, bm))`,
-where `bm` is an [`diffrax.AbstractBrownianPath`][] and the same values of `gammma` and `u` are passed to both terms.
+where `bm` is an [`diffrax_extensions.AbstractBrownianPath`][] and the same values of `gammma` and `u` are passed to both terms.
 
-::: diffrax.UnderdampedLangevinDriftTerm
+::: diffrax_extensions.UnderdampedLangevinDriftTerm
     selection:
         members:
             - __init__
 
-::: diffrax.UnderdampedLangevinDiffusionTerm
+::: diffrax_extensions.UnderdampedLangevinDiffusionTerm
     selection:
         members:
             - __init__