mirror of
https://github.com/DifferentiableUniverseInitiative/JaxPM.git
synced 2025-04-07 12:20:54 +00:00
use jax interp instead of jax_cosmo
This commit is contained in:
parent
f3b431aa74
commit
2ad035a75b
1 changed files with 1 additions and 1 deletions
|
@ -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
|
||||
|
||||
|
||||
|
|
Loading…
Add table
Reference in a new issue