fix: mypy errors with numpy 2.2

alexfikl · Dec 10, 2024 · 56c86dc · 56c86dc
1 parent fe994c4
commit 56c86dc
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/examples/burgers-ssweno-periodic-numpy.py b/examples/burgers-ssweno-periodic-numpy.py
@@ -435,7 +435,7 @@ def burgers_rhs(
 
     fssw[0] = fssw[-1]
 
-    return -(fssw[1:] - fssw[:-1]) / dx
+    return -(fssw[1:] - fssw[:-1]) / dx  # type: ignore[operator]
 
 
 # }}}

diff --git a/src/pyshocks/timestepping.py b/src/pyshocks/timestepping.py
@@ -109,7 +109,7 @@ def step(
     """
     if tfinal is None:
         # NOTE: can't set this to jnp.inf because we have a debug check for it
-        tfinal = jnp.finfo(u0.dtype).max  # type: ignore[no-untyped-call]
+        tfinal = float(jnp.finfo(u0.dtype).max)  # type: ignore[no-untyped-call]
 
     m = 0
     t = jnp.array(tstart, dtype=u0.dtype)

diff --git a/src/pyshocks/tools.py b/src/pyshocks/tools.py
@@ -264,7 +264,7 @@ def satisfied(
 
         _, error = self._history
         if atol is None:
-            atol = 1.0e2 * jnp.finfo(error.dtype).eps  # type: ignore[no-untyped-call]
+            atol = 1.0e2 * float(jnp.finfo(error.dtype).eps)  # type: ignore[no-untyped-call]
 
         return bool(self.estimated_order >= (order - slack) or jnp.max(error) < atol)
-Original file line number
+Diff line change
@@ Expand Up / @@ -435,7 +435,7 @@ def burgers_rhs( @@
         fssw[0] = fssw[-1]
-        return -(fssw[1:] - fssw[:-1]) / dx
+        return -(fssw[1:] - fssw[:-1]) / dx  # type: ignore[operator]
     # }}}
@@ Expand Down @@