fix: missed an import

beckermr · beckermr · commit 1aeb448e4378 · 2026-05-21T07:13:36.000-05:00
diff --git a/jax_galsim/photon_array.py b/jax_galsim/photon_array.py
@@ -1,5 +1,6 @@
 from contextlib import contextmanager
 
+import equinox
 import galsim as _galsim
 import jax
 import jax.numpy as jnp
