From 2ad035a75b86960f1402ee22cd9b9a053d0f8ed1 Mon Sep 17 00:00:00 2001 From: Wassim KABALAN Date: Wed, 30 Oct 2024 01:56:04 +0100 Subject: [PATCH] use jax interp instead of jax_cosmo --- jaxpm/growth.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/jaxpm/growth.py b/jaxpm/growth.py index 80e6698..ec248f3 100644 --- a/jaxpm/growth.py +++ b/jaxpm/growth.py @@ -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