fix #9823

google · Mar 16, 2022 · b0b4dfd · b0b4dfd
1 parent 4cdc25f
commit b0b4dfd
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/jax/core.py b/jax/core.py
@@ -868,7 +868,8 @@ def f(x):
       with jax.ensure_compile_time_eval():
         y = jnp.sin(3.0)
         z = jnp.sin(y)
-      if z > 0:  # the value of z is availble and can be used in control flow
+        z_positive = z > 0
+      if z_positive:  # z_positive is usable in Python control flow
         return jnp.sin(x)
       else:
         return jnp.cos(x)