fix: use proper array ref

beckermr · beckermr · commit 88c1c7b546ee · 2026-05-15T08:17:50.000-05:00
diff --git a/jax_galsim/core/utils.py b/jax_galsim/core/utils.py
@@ -25,7 +25,7 @@
 )
 CONST_TYPES_WITH_JAX = CONST_TYPES + (
     jax.Array,
-    jnp.array,
+    jnp.ndarray,
     jnp.int8,
     jnp.int16,
     jnp.int32,

Original file line number	Diff line number	Diff line change
`@@ -25,7 +25,7 @@`
`25`	`25`	`)`
`26`	`26`	`CONST_TYPES_WITH_JAX = CONST_TYPES + (`
`27`	`27`	`jax.Array,`
`28`		`- jnp.array,`
	`28`	`+ jnp.ndarray,`
`29`	`29`	`jnp.int8,`
`30`	`30`	`jnp.int16,`
`31`	`31`	`jnp.int32,`