Cleaner solution to b4151d9

PlasmaControl · Sep 16, 2024 · 8d9b605 · 8d9b605
1 parent 52adba9
commit 8d9b605
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 7 deletions.
diff --git a/desc/integrals/interp_utils.py b/desc/integrals/interp_utils.py
@@ -203,13 +203,8 @@ def _root_cubic(C, sentinel, eps, distinct):
 
     def irreducible(Q, R, b, mask):
         # Three irrational real roots.
-        theta = jnp.arccos(
-            jnp.clip(
-                R / jnp.sqrt(jnp.where(mask, Q**3, 1.0)),
-                -1.0 + 0.01 * eps,
-                +1.0 - 0.01 * eps,
-            )
-        )
+        theta = R / jnp.sqrt(jnp.where(mask, Q**3, 1.0))
+        theta = jnp.arccos(jnp.where(jnp.abs(theta) < 1.0, theta, 0.0))
         return jnp.moveaxis(
             -2
             * jnp.sqrt(Q)

diff --git a/tests/baseline/test_bounce1d_checks.png b/tests/baseline/test_bounce1d_checks.png