Cosmology, Cache, and Configuration data model

.github/workflows/pythonpackage.yml

@@ -15,28 +15,27 @@ jobs:
     runs-on: ubuntu-latest
     strategy:
       matrix:
-        python-version: [3.6, 3.7, 3.8]
+        python-version: ["3.7", "3.8", "3.9", "3.10"]
 
     steps:
     - uses: actions/checkout@v2
 
     - uses: conda-incubator/setup-miniconda@v2
       with:
+        miniforge-variant: Mambaforge
+        use-mamba: true
         python-version: ${{ matrix.python-version }}
-        channels: conda-forge,defaults
-        channel-priority: strict
-        show-channel-urls: true
         auto-update-conda: true
+        show-channel-urls: true
 
-    - name: Install dependencies
+    - name: Install package and dependencies
       shell: bash -l {0}
       run: |
         conda config --set always_yes yes
-        conda install pytest pip 
-        conda install -c conda-forge pyccl
+        mamba install pytest pytest-xdist pyccl
         pip install .
-  
+
     - name: Test with pytest
       shell: bash -l {0}
       run: |
-        pytest
+        pytest -n auto

jax_cosmo/angular_cl.py

@@ -70,7 +70,7 @@ def angular_cl(
     def cl(ell):
         def integrand(a):
             # Step 1: retrieve the associated comoving distance
-            chi = bkgrd.radial_comoving_distance(cosmo, a)
+            _, chi = bkgrd.radial_comoving_distance(cosmo, a)
 
             # Step 2: get the power spectrum for this combination of chi and a
             k = (ell + 0.5) / np.clip(chi, 1.0)