When using JAX don't import Numpy

Jammy2211 / PyAutoGalaxy

PyAutoGalaxy: Open-Source Multiwavelength Galaxy Structure & Morphology

https://pyautogalaxy.readthedocs.io/

MIT License

26 stars 13 forks source link

Open CKrawczyk opened 2 weeks ago

CKrawczyk commented 2 weeks ago

Switches out numpy for jax.numpy. A very simple fix that makes radial profiles work as expected.