fix: use any everywhere

Author: beckermr

jax_galsim/gsobject.py

@@ -629,10 +629,13 @@ def drawImage(
             raise TypeError("image is not an Image instance", image)
 
         # Make sure (gain, area, exptime) have valid values:
-        gain = equinox.error_if(jnp.array(gain), gain <= 0.0, "Invalid gain <= 0.")
-        area = equinox.error_if(jnp.array(area), area <= 0.0, "Invalid area <= 0.")
+        gain = jnp.array(gain)
+        gain = equinox.error_if(gain, jnp.any(gain <= 0.0), "Invalid gain <= 0.")
+        area = jnp.array(area)
+        area = equinox.error_if(area, jnp.any(area <= 0.0), "Invalid area <= 0.")
+        exptime = jnp.array(exptime)
         exptime = equinox.error_if(
-            jnp.array(exptime), exptime <= 0.0, "Invalid exptime <= 0."
+            exptime, jnp.any(exptime <= 0.0), "Invalid exptime <= 0."
         )
 
         if method == "phot" and save_photons and maxN is not None:
@@ -1228,7 +1231,8 @@ def drawPhot(
         elif not isinstance(sensor, Sensor):
             raise TypeError("The sensor provided is not a Sensor instance")
 
-        gain = equinox.error_if(jnp.array(gain), gain <= 0.0, "Invalid gain <= 0.")
+        gain = jnp.array(gain)
+        gain = equinox.error_if(gain, jnp.any(gain <= 0.0), "Invalid gain <= 0.")
 
         if n_photons is not None:
             # n_photons is the length of an array so it is a python int and

jax_galsim/integ.py

@@ -75,6 +75,6 @@ def _base_integration():
 
     return equinox.error_if(
         val,
-        status != 0,
+        jnp.any(status != 0),
         "`jax_galsim.int1d` failed to converge!",
     )

jax_galsim/interpolatedimage.py

@@ -515,9 +515,10 @@ def __init__(
                 )
 
         if calculate_stepk or calculate_maxk or flux is not None:
+            image.array = jnp.array(image.array)
             image.array = equinox.error_if(
-                jnp.array(image.array),
-                image.array.sum() == 0.0,
+                image.array,
+                jnp.any(image.array.sum() == 0.0),
                 "This input image has zero total flux. It does not define a "
                 "valid surface brightness profile.",
             )

jax_galsim/shear.py

@@ -53,7 +53,7 @@ def __init__(self, *args, **kwargs):
             self._g = g1 + 1j * g2
             self._g = equinox.error_if(
                 self._g,
-                jnp.abs(self._g) > 1.0,
+                jnp.any(jnp.abs(self._g) > 1.0),
                 "Requested shear exceeds 1.",
             )
 
@@ -64,7 +64,7 @@ def __init__(self, *args, **kwargs):
             absesq = e1**2 + e2**2
             absesq = equinox.error_if(
                 absesq,
-                absesq > 1.0,
+                jnp.any(absesq > 1.0),
                 "Requested distortion exceeds 1.",
             )
             self._g = (e1 + 1j * e2) * self._e2g(absesq)
@@ -91,7 +91,7 @@ def __init__(self, *args, **kwargs):
             g = jnp.array(kwargs.pop("g"))
             g = equinox.error_if(
                 g,
-                g > 1 or g < 0,
+                jnp.any((g > 1) | (g < 0)),
                 "Requested |shear| is outside [0,1].",
             )
             self._g = g * jnp.exp(2j * beta.rad)
@@ -110,7 +110,7 @@ def __init__(self, *args, **kwargs):
             e = jnp.array(kwargs.pop("e"))
             e = equinox.error_if(
                 e,
-                (e > 1) | (e < 0),
+                jnp.any((e > 1) | (e < 0)),
                 "Requested distortion is outside [0,1].",
             )
             self._g = self._e2g(e**2) * e * jnp.exp(2j * beta.rad)
@@ -129,7 +129,7 @@ def __init__(self, *args, **kwargs):
             eta = jnp.array(kwargs.pop("eta"))
             eta = equinox.error_if(
                 eta,
-                eta < 0,
+                jnp.any(eta < 0),
                 "Requested eta is below 0.",
             )
             self._g = self._eta2g(eta) * eta * jnp.exp(2j * beta.rad)
@@ -148,7 +148,7 @@ def __init__(self, *args, **kwargs):
             q = jnp.array(kwargs.pop("q"))
             q = equinox.error_if(
                 q,
-                (q <= 0) | (q > 1),
+                jnp.any((q <= 0) | (q > 1)),
                 "Cannot use requested axis ratio.",
             )
             eta = -jnp.log(q)