Wassim Kabalan
09d82283f5
Merge remote-tracking branch 'origin/main' into 41-spherical-lensing
2025-06-29 11:32:30 +02:00
Wassim KABALAN
7d76573701
Replace np.power 0.5 by np.sqrt ( #43 )
...
Tests / run_tests (3.10) (push) Failing after 2m35s
Tests / run_tests (3.11) (push) Failing after 2m37s
Code Formatting / formatting (push) Successful in 3m1s
Tests / run_tests (3.12) (push) Failing after 1m16s
* Switch **2 by np.sqrt
* format
---------
Co-authored-by: Francois Lanusse <fr.eiffel@gmail.com>
2025-06-29 10:51:45 +02:00
Francois Lanusse
9d0b047f06
fixing formatting
2025-06-28 23:21:31 +02:00
Francois Lanusse
1fcfc2aef2
Merge branch 'main' into 41-spherical-lensing
2025-06-28 23:12:39 +02:00
Wassim KABALAN
6693e5c725
Fix sharding error ( #37 )
...
Code Formatting / formatting (push) Failing after 4m30s
Tests / run_tests (3.10) (push) Failing after 1m41s
Tests / run_tests (3.11) (push) Failing after 1m42s
Tests / run_tests (3.12) (push) Failing after 1m15s
* Use cosmo as arg for the ODE function
* Update examples
* format
* notebook update
* fix tests
* add correct annotations for weights in painting and warning for cic_paint in distributed pm
* update test_against_fpm
* update distributed tests and add jacfwd jacrev and vmap tests
* format
* add Caveats to notebook readme
* final touches
* update Growth.py to allow using FastPM solver
* fix 2D painting when input is (X , Y , 2) shape
* update cic read halo size and notebooks examples
* Allow env variable control of caching in growth
* Format
* update test jax version
* update notebooks/03-MultiGPU_PM_Halo.ipynb
* update numpy install in wf
* update tolerance :)
* reorganize install in test workflow
* update tests
* add mpi4py
* update tests.yml
* update tests
* update wf
* format
* make normal_field signature consistent with jax.random.normal
* update by default normal_field dtype to match JAX
* format
* debug test workflow
* format
* debug test workflow
* updating tests
* fix accuracy
* fixed tolerance
* adding caching
* Update conftest.py
* Update tolerance and precision settings in distributed PM tests
* revererting back changes to growth.py
---------
Co-authored-by: Francois Lanusse <fr.eiffel@gmail.com>
Co-authored-by: Francois Lanusse <EiffL@users.noreply.github.com>
2025-06-28 23:07:31 +02:00
Wassim Kabalan
f6d547e31f
Add Spherical lensing example
2025-06-28 19:25:14 +02:00
Wassim Kabalan
2d21985279
Update tolerance and precision settings in distributed PM tests
2025-06-28 12:08:16 +02:00
Francois Lanusse
e666aada42
Update conftest.py
2025-06-28 02:57:20 +02:00
Francois Lanusse
844cace3d7
adding caching
2025-06-28 02:41:16 +02:00
Francois Lanusse
37976c70d8
fixed tolerance
2025-06-28 02:36:37 +02:00
Francois Lanusse
0c75158043
fix accuracy
2025-06-28 02:21:19 +02:00
Francois Lanusse
627be7a764
updating tests
2025-06-28 01:36:17 +02:00
Wassim Kabalan
7623e60581
debug test workflow
2025-06-12 15:28:03 +02:00
Wassim Kabalan
d874790543
format
2025-06-12 15:20:20 +02:00
Wassim Kabalan
187cf5c4ba
debug test workflow
2025-06-12 15:19:36 +02:00
Wassim Kabalan
67a80e1041
format
2025-06-12 14:51:24 +02:00
Wassim Kabalan
e1a8134b8e
update by default normal_field dtype to match JAX
2025-06-12 14:46:27 +02:00
Wassim Kabalan
f391a79211
make normal_field signature consistent with jax.random.normal
2025-06-09 19:35:40 +02:00
Wassim Kabalan
5807e1d3f4
format
2025-06-08 15:37:06 +02:00
Wassim Kabalan
a7fcba0e1f
update wf
2025-06-08 15:35:57 +02:00
Wassim Kabalan
c6a7dd4e4e
update tests
2025-06-08 15:27:32 +02:00
Wassim Kabalan
e0ba85fb58
update tests.yml
2025-06-08 12:03:44 +02:00
Wassim Kabalan
12eddc4e6a
add mpi4py
2025-06-08 11:48:11 +02:00
Wassim Kabalan
0eb4c371e3
update tests
2025-06-08 11:43:15 +02:00
Wassim Kabalan
3be619a2db
reorganize install in test workflow
2025-06-08 11:36:06 +02:00
Wassim Kabalan
f0b849cf5f
update tolerance :)
2025-06-08 11:35:57 +02:00
Wassim Kabalan
995cc4c78c
update numpy install in wf
2025-06-08 11:26:32 +02:00
Wassim Kabalan
7b7205e3b3
update notebooks/03-MultiGPU_PM_Halo.ipynb
2025-06-08 11:25:51 +02:00
Wassim Kabalan
6aacd81bd6
update test jax version
2025-06-08 10:56:34 +02:00
Wassim Kabalan
49c93aacf6
Format
2025-06-08 10:45:20 +02:00
Wassim Kabalan
41ae41ace3
Allow env variable control of caching in growth
2025-06-08 10:45:04 +02:00
Wassim Kabalan
e7112e0c25
update cic read halo size and notebooks examples
2025-06-07 19:26:37 +02:00
Wassim Kabalan
d4049e5db4
fix 2D painting when input is (X , Y , 2) shape
2025-05-09 22:07:53 +02:00
Wassim Kabalan
20ace41d32
update Growth.py to allow using FastPM solver
2025-05-08 15:08:28 +02:00
Wassim Kabalan
e1daa8cba4
final touches
2025-02-28 14:03:33 +01:00
Wassim Kabalan
4e4d3745f0
add Caveats to notebook readme
2025-02-28 13:48:08 +01:00
Wassim Kabalan
9f494da317
format
2025-02-28 13:47:43 +01:00
Wassim Kabalan
580387ce1c
update distributed tests and add jacfwd jacrev and vmap tests
2025-02-28 13:47:22 +01:00
Wassim Kabalan
eb0778aa24
update test_against_fpm
2025-02-28 13:47:00 +01:00
Wassim Kabalan
91d3292923
add correct annotations for weights in painting and warning for cic_paint in distributed pm
2025-02-28 13:46:41 +01:00
Wassim Kabalan
f8325b1c67
fix tests
2025-02-28 10:06:09 +01:00
Wassim Kabalan
f3b8f4160e
notebook update
2025-02-28 09:59:04 +01:00
Wassim Kabalan
b43cb373a0
format
2025-02-28 09:56:00 +01:00
Wassim Kabalan
51ee4dd937
update examples
2025-02-28 09:54:53 +01:00
Wassim Kabalan
8e0f300572
Update examples
2025-02-26 14:32:47 +01:00
Wassim Kabalan
0b08c6f59a
Use cosmo as arg for the ODE function
2025-02-26 14:05:25 +01:00
Wassim KABALAN
cb2a7ab17f
Fix pfft gradients ( #34 )
...
* update jaxdecomp version and test gradients
* Prepare for DTO tests
* format
2024-12-22 12:47:42 -05:00
Francois Lanusse
d81a2529e7
minor typo fix
2024-12-21 15:28:20 -05:00
Francois Lanusse
15f2fb1ee6
adding notice
2024-12-21 15:26:53 -05:00
Francois Lanusse
ae0f439ae4
fixing formatting of notebook
2024-12-21 13:14:42 -05:00
