Test against Python 3.13 (#2058)

* test python 3.13

* missing changes

Author: juanitorduz

.github/workflows/ci.yml

@@ -17,7 +17,7 @@ jobs:
     runs-on: ubuntu-latest
     strategy:
       matrix:
-        python-version: ["3.9","3.10","3.12"]
+        python-version: ["3.9","3.10","3.13"]
 
     steps:
     - uses: actions/checkout@v2
@@ -56,7 +56,7 @@ jobs:
     needs: lint
     strategy:
       matrix:
-        python-version: ["3.9", "3.10"]
+        python-version: ["3.9", "3.13"]
 
     steps:
     - uses: actions/checkout@v2
@@ -82,7 +82,7 @@ jobs:
       run: |
         JAX_ENABLE_X64=1 pytest -vs test/test_distributions.py -k "powerLaw or Dagum"
     - name: Test tracer leak
-      if: matrix.python-version == '3.10'
+      if: matrix.python-version == '3.13'
       env:
         JAX_CHECK_TRACER_LEAKS: 1
       run: |
@@ -94,7 +94,7 @@ jobs:
         pytest -vs test/test_distributions.py::test_mean_var -k Gompertz
 
     - name: Coveralls
-      if: github.repository == 'pyro-ppl/numpyro' && matrix.python-version == '3.10'
+      if: github.repository == 'pyro-ppl/numpyro' && matrix.python-version == '3.13'
       uses: coverallsapp/github-action@v2
       with:
         github-token: ${{ secrets.GITHUB_TOKEN }}
@@ -108,7 +108,7 @@ jobs:
     needs: lint
     strategy:
       matrix:
-        python-version: ["3.9", "3.10"]
+        python-version: ["3.9", "3.13"]
 
     steps:
     - uses: actions/checkout@v2
@@ -147,7 +147,7 @@ jobs:
       run: |
         JAX_ENABLE_X64=1 pytest -vs test/contrib/test_nested_sampling.py
     - name: Coveralls
-      if: github.repository == 'pyro-ppl/numpyro' && matrix.python-version == '3.10'
+      if: github.repository == 'pyro-ppl/numpyro' && matrix.python-version == '3.13'
       uses: coverallsapp/github-action@v2
       with:
         github-token: ${{ secrets.GITHUB_TOKEN }}
@@ -161,7 +161,7 @@ jobs:
     needs: lint
     strategy:
       matrix:
-        python-version: ["3.10"]
+        python-version: ["3.13"]
 
     steps:
     - uses: actions/checkout@v2
@@ -181,7 +181,7 @@ jobs:
       run: |
         CI=1 XLA_FLAGS="--xla_force_host_platform_device_count=2" pytest -vs -k test_example
     - name: Coveralls
-      if: github.repository == 'pyro-ppl/numpyro' && matrix.python-version == '3.10'
+      if: github.repository == 'pyro-ppl/numpyro' && matrix.python-version == '3.13'
       uses: coverallsapp/github-action@v2
       with:
         github-token: ${{ secrets.GITHUB_TOKEN }}