guilhem_lavaux/JaxPM
0
0
Fork 1
mirror of https://github.com/DifferentiableUniverseInitiative/JaxPM.git synced 2025-06-28 08:01:12 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Update conftest.py

Browse source
This commit is contained in:
Francois Lanusse 2025-06-28 02:57:20 +02:00 • committed by GitHub
parent 844cace3d7
commit e666aada42
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
tests/conftest.py
View file

@ -93,7 +93,7 @@ def fpm_initial_conditions(cosmo, particle_mesh):
def pk_fn(x): def pk_fn(x):
return jnp.interp(x.reshape([-1]), k, pk).reshape(x.shape) return jnp.interp(x.reshape([-1]), k, pk).reshape(x.shape)
whitec = particle_mesh.generate_whitenoise(123, whitec = particle_mesh.generate_whitenoise(42,
type='complex', type='complex',
unitary=False) unitary=False)
lineark = whitec.apply(lambda k, v: pk_fn(sum(ki**2 for ki in k)**0.5)**0.5 lineark = whitec.apply(lambda k, v: pk_fn(sum(ki**2 for ki in k)**0.5)**0.5