guilhem_lavaux/JaxPM
0
0
Fork 1
mirror of https://github.com/DifferentiableUniverseInitiative/JaxPM.git synced 2025-06-29 16:41:11 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Replace np.power 0.5 by np.sqrt (#43)
Some checks failed
Tests / run_tests (3.10) (push) Failing after 2m35s
Details
Tests / run_tests (3.11) (push) Failing after 2m37s
Details
Code Formatting / formatting (push) Successful in 3m1s
Details
Tests / run_tests (3.12) (push) Failing after 1m16s
Details

Browse source 
* Switch **2  by np.sqrt

* format

---------

Co-authored-by: Francois Lanusse <fr.eiffel@gmail.com>
This commit is contained in:
Wassim KABALAN 2025-06-29 10:51:45 +02:00 • committed by GitHub
parent 6693e5c725
commit 7d76573701
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
4 changed files with 6 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

2
jaxpm/pm.py
View file

@ -139,7 +139,7 @@ def linear_field(mesh_shape, box_size, pk, seed, sharding=None):
pkmesh = pk(kmesh) * (mesh_shape[0] * mesh_shape[1] * mesh_shape[2]) / (
box_size[0] * box_size[1] * box_size[2])
field = field * (pkmesh)**0.5
field = field * jnp.sqrt(pkmesh)
field = ifft3d(field)
return field