Fix benign bug in midpoint evaluation. (#17)

Resolves #16
f0uriest · Dec 3, 2024 · 0cb837c · 0cb837c
2 parents 48fe56b + fa14060
commit 0cb837c
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/quadax/adaptive.py b/quadax/adaptive.py
@@ -422,7 +422,7 @@ def adaptive_quadrature(
     epsrel = setdefault(epsrel, jnp.sqrt(jnp.finfo(jnp.array(1.0)).eps))
     fun, interval = map_interval(fun, interval)
     vfunc = wrap_func(fun, args)
-    f = jax.eval_shape(vfunc, (interval[0] + interval[-1] / 2))
+    f = jax.eval_shape(vfunc, (interval[0] + interval[-1]) / 2)
     epmach = jnp.finfo(f.dtype).eps
     shape = f.shape
 

diff --git a/quadax/romberg.py b/quadax/romberg.py
@@ -97,7 +97,7 @@ def romberg(
     fun, interval = map_interval(fun, interval)
     vfunc = wrap_func(fun, args)
     a, b = interval
-    f = jax.eval_shape(vfunc, (a + b / 2))
+    f = jax.eval_shape(vfunc, (a + b) / 2)
 
     result = jnp.zeros((divmax + 1, divmax + 1, *f.shape), f.dtype)
     result = result.at[0, 0].set(vfunc(a) + vfunc(b))