Develop

.github/workflows/tests.yml

@@ -152,6 +152,6 @@ jobs:
       - name: Test imports
         run: |
           poetry run python -c "import jaxeffort; print('✓ jaxeffort import successful')"
-          poetry run python -c "from jaxeffort import W0WaCDMCosmology, E_z, D_z, f_z; print('✓ background functions import successful')"
+          poetry run python -c "from jaxeffort import w0waCDMCosmology, E_z, D_z, f_z; print('✓ background functions import successful')"
           poetry run python -c "from jaxeffort import MultipoleEmulators; print('✓ emulator classes import successful')"
           poetry run python -c "from jaxeffort import load_multipole_emulator; print('✓ loader functions import successful')"

Artifacts.toml

@@ -0,0 +1,34 @@
+# Artifacts.toml for jaxeffort trained emulators
+#
+# This file declares the available trained emulators for jaxeffort.
+# Compatible with Julia's Pkg.Artifacts system via pyartifacts.
+
+[pybird_mnuw0wacdm]
+git-tree-sha1 = "d309b571f5693718c8612d387820a409479fe50688d4c46c87ba8662c6acc09b"
+lazy = true
+description = "PyBird emulator for massive neutrinos, w0wa CDM cosmology"
+has_noise = false
+
+    [[pybird_mnuw0wacdm.download]]
+    url = "https://zenodo.org/records/17436464/files/trained_effort_pybird_mnuw0wacdm.tar.xz?download=1"
+    sha256 = "d309b571f5693718c8612d387820a409479fe50688d4c46c87ba8662c6acc09b"
+
+[velocileptors_lpt_mnuw0wacdm]
+git-tree-sha1 = "4f942141ecdfe8a71b03f514a47103fe9a5ae3e3c3b46963c7627b4e196ca579"
+lazy = true
+description = "Velocileptors LPT emulator for massive neutrinos, w0wa CDM cosmology"
+has_noise = false
+
+    [[velocileptors_lpt_mnuw0wacdm.download]]
+    url = "https://zenodo.org/records/17635885/files/trained_effort_velocileptors_lpt_mnuw0wacdm.tar.xz?download=1"
+    sha256 = "4f942141ecdfe8a71b03f514a47103fe9a5ae3e3c3b46963c7627b4e196ca579"
+
+[velocileptors_rept_mnuw0wacdm]
+git-tree-sha1 = "33dfa33b1191928198d17f263bae77621d8f46cff25281812cbc2cdddff092d9"
+lazy = true
+description = "Velocileptors REPT emulator for massive neutrinos, w0wa CDM cosmology"
+has_noise = false
+
+    [[velocileptors_rept_mnuw0wacdm.download]]
+    url = "https://zenodo.org/records/17635853/files/trained_effort_velocileptors_rept_mnuw0wacdm.tar.xz?download=1"
+    sha256 = "33dfa33b1191928198d17f263bae77621d8f46cff25281812cbc2cdddff092d9"

docs/api.md

@@ -21,7 +21,7 @@ Extension of MultipoleEmulators with stochastic noise terms.
 
 ## Cosmology Classes
 
-### W0WaCDMCosmology
+### w0waCDMCosmology
 
 w₀wₐCDM cosmology with massive neutrinos support.
 

docs/contributing.md

@@ -130,7 +130,7 @@ def test_multipole_emulator_loading():
 def test_invalid_parameters():
     """Test handling of invalid parameters."""
     with pytest.raises(ValueError):
-        jaxeffort.W0WaCDMCosmology(h=-0.5)  # Invalid Hubble parameter
+        jaxeffort.w0waCDMCosmology(h=-0.5)  # Invalid Hubble parameter
 ```
 
 ## Reporting Issues

docs/index.md

@@ -46,7 +46,7 @@ cosmo_params = jnp.array([z, ln10As, ns, H0, ombh2, omch2, Mnu, w0, wa])
 bias_params = jnp.array([b1, b2, b3, b4, b5, b6, b7, f])
 
 # Compute growth factor
-cosmo = jaxeffort.W0WaCDMCosmology(...)
+cosmo = jaxeffort.w0waCDMCosmology(...)
 D = cosmo.D_z(z)
 
 # Get multipoles

docs/usage_examples.md

@@ -47,7 +47,7 @@ bias_params = jnp.array([
 ])
 
 # Setup cosmology for growth factor calculation
-cosmo = jaxeffort.W0WaCDMCosmology(
+cosmo = jaxeffort.w0waCDMCosmology(
     ln10As=cosmo_params[1],
     ns=cosmo_params[2],
     h=cosmo_params[3]/100,
@@ -190,7 +190,7 @@ cosmo_lcdm = jaxeffort.LCDMCosmology(
 )
 
 # w0waCDM cosmology
-cosmo_w0wa = jaxeffort.W0WaCDMCosmology(
+cosmo_w0wa = jaxeffort.w0waCDMCosmology(
     ln10As=3.1,
     ns=0.96,
     h=0.67,

generate_doc_plots.py

@@ -66,7 +66,7 @@ def generate_multipoles_plot():
         b = jnp.ones(8)
 
         # Compute D(z) using cosmology
-        cosmo = jaxeffort.W0WaCDMCosmology(
+        cosmo = jaxeffort.w0waCDMCosmology(
             ln10As=theta[1],
             ns=theta[2],
             h=theta[3]/100,
@@ -161,7 +161,7 @@ def generate_jacobian_multipoles_plot():
         b = jnp.ones(8)
 
         # Compute D(z)
-        cosmo = jaxeffort.W0WaCDMCosmology(
+        cosmo = jaxeffort.w0waCDMCosmology(
             ln10As=theta[1],
             ns=theta[2],
             h=theta[3]/100,
@@ -254,7 +254,7 @@ def generate_comparison_plot():
         b = jnp.ones(8)
 
         # Compute D(z)
-        cosmo = jaxeffort.W0WaCDMCosmology(
+        cosmo = jaxeffort.w0waCDMCosmology(
             ln10As=theta[1],
             ns=theta[2],
             h=theta[3]/100,