Merge branch 'main' into typing-inits-cleanup

Author: beckermr

.github/workflows/python_package.yaml

@@ -40,8 +40,7 @@ jobs:
       - name: Install dependencies
         run: |
           python -m pip install --upgrade pip
-          python -m pip install pytest pytest-codspeed pytest-randomly pytest-split pytest-retry
-          python -m pip install .
+          python -m pip install .[dev]
 
       - name: Restore test durations
         uses: actions/cache/restore@cdf6c1fa76f9f475f3d7449005a359c84ca0f306  # v5.0.3
@@ -83,7 +82,8 @@ jobs:
             --durations-path=.test_durations.${{ matrix.group }} \
             --splitting-algorithm least_duration \
             --clean-durations \
-            --retries 1
+            --retries 1 \
+            -n 2
 
       - name: Upload test durations
         uses: actions/upload-artifact@v7

pyproject.toml

@@ -13,7 +13,7 @@ readme          = "README.md"
 dependencies    = ["numpy >=1.18.0", "galsim >=2.7.0", "jax >=0.6.0", "astropy >=2.0", "quadax", "equinox"]
 
 [project.optional-dependencies]
-dev  = ["pytest", "pytest-codspeed"]
+dev  = ["pytest", "pytest-codspeed", "pytest-randomly", "pytest-split", "pytest-retry", "pytest-xdist"]
 docs = ["sphinx>=7.0", "furo>=2024.1.29", "sphinx-design>=0.5", "sphinx-copybutton"]
 
 [project.urls]