Import ShapedArray from jax.core
instead for jax.abstract_arrays
#91
Merged
dfm merged 2 commits intoexoplanet-dev:mainfrom vandalt:shapedarray-import-fixOct 5, 2023
+2-2
jax.core
instead for jax.abstract_arrays
#91