fix: need to ensure images always hold jax arrays

Author: beckermr

jax_galsim/image.py

@@ -102,7 +102,7 @@ def __init__(self, *args, **kwargs):
                 )
         else:
             if "array" in kwargs:
-                array = kwargs.pop("array")
+                array = jnp.array(kwargs.pop("array"))
                 array, xmin, ymin = self._get_xmin_ymin(
                     array, kwargs, check_bounds=_check_bounds
                 )

jax_galsim/interpolatedimage.py

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