Skip to content

Commit c892e04

Browse files
committed
fix: raise if we do not have array as kwarg
1 parent 21ffe09 commit c892e04

1 file changed

Lines changed: 8 additions & 1 deletion

File tree

jax_galsim/image.py

Lines changed: 8 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -102,7 +102,14 @@ def __init__(self, *args, **kwargs):
102102
)
103103
else:
104104
if "array" in kwargs:
105-
array = jnp.array(kwargs.pop("array"))
105+
array = kwargs.pop("array")
106+
if has_tracers(array) or isinstance(array, jnp.ndarray):
107+
pass
108+
elif isinstance(array, np.ndarray):
109+
array = jnp.array(cast_numpy_array_to_native_byte_order(array))
110+
else:
111+
raise TypeError("Unable to parse %s as an array." % array)
112+
106113
array, xmin, ymin = self._get_xmin_ymin(
107114
array, kwargs, check_bounds=_check_bounds
108115
)

0 commit comments

Comments
 (0)