use jax interp instead of jax_cosmo

This commit is contained in:
Wassim KABALAN 2024-10-30 01:56:04 +01:00
parent f3b431aa74
commit 2ad035a75b

View file

@ -1,6 +1,6 @@
import jax.numpy as np
from jax.numpy import interp
from jax_cosmo.background import *
from jax_cosmo.scipy.interpolate import interp
from jax_cosmo.scipy.ode import odeint