added some missing decimal points

Herculens · Jun 6, 2024 · f4069b6 · f4069b6
1 parent b802a0f
commit f4069b6
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/herculens/Util/jax_util.py b/herculens/Util/jax_util.py
@@ -60,7 +60,7 @@ def R_omega(z, t, q, nmax):
     @jit
     def body_fun(i, val):
         # Currrent term in the series is proportional to the previous
-        ratio = (2. * i + t - 2) / (2. * i - t + 2.)
+        ratio = (2. * i + t - 2.) / (2. * i - t + 2.)
         val[1] = -f * ei2phi * ratio * val[1]
         # Adds the current term to the partial sum
         val[0] += val[1]
@@ -89,7 +89,7 @@ def omega_real(x, y, t, q, nmax):
     Cs, Ss = jnp.cos(phi), jnp.sin(phi)
     Cs2, Ss2 = jnp.cos(2 * phi), jnp.sin(2 * phi)
     def update(n, val):
-        prefac = -f * (2 * n - (2 - t)) / (2 * n + (2 - t))
+        prefac = -f * (2. * n - (2. - t)) / (2. * n + (2. - t))
         last_x, last_y, fx, fy = val
         last_x, last_y = prefac * (Cs2 * last_x - Ss2 * last_y), prefac * (
                 Ss2 * last_x + Cs2 * last_y