Test on Python 3.12

Author: ricardoV94

.github/workflows/tests.yml

@@ -48,7 +48,7 @@ jobs:
       matrix:
         os: [ubuntu-20.04]
         floatx: [float64]
-        python-version: ["3.11"]
+        python-version: ["3.12"]
         test-subset:
           - |
             tests/test_util.py
@@ -263,7 +263,7 @@ jobs:
       matrix:
         os: [macos-latest]
         floatx: [float64]
-        python-version: ["3.10"]
+        python-version: ["3.12"]
         test-subset:
           - |
             tests/sampling/test_parallel.py
@@ -342,7 +342,7 @@ jobs:
       matrix:
         os: [ubuntu-20.04]
         floatx: [float64]
-        python-version: ["3.11"]
+        python-version: ["3.12"]
         test-subset:
           - tests/sampling/test_jax.py tests/sampling/test_mcmc_external.py
       fail-fast: false
@@ -410,7 +410,7 @@ jobs:
       matrix:
         os: [windows-latest]
         floatx: [float32]
-        python-version: ["3.11"]
+        python-version: ["3.12"]
         test-subset:
         - tests/sampling/test_mcmc.py tests/ode/test_ode.py tests/ode/test_utils.py tests/distributions/test_transform.py
       fail-fast: false

conda-envs/environment-jax.yml

@@ -13,8 +13,8 @@ dependencies:
 - h5py>=2.7
 # Jaxlib version must not be greater than jax version!
 - blackjax>=1.0.0
-- jaxlib==0.4.14
-- jax==0.4.16
+- jaxlib==0.4.23
+- jax==0.4.23
 - libblas=*=*mkl
 - mkl-service
 - numpy>=1.15.0

setup.py

@@ -33,6 +33,7 @@
     "Programming Language :: Python :: 3.9",
     "Programming Language :: Python :: 3.10",
     "Programming Language :: Python :: 3.11",
+    "Programming Language :: Python :: 3.12",
     "License :: OSI Approved :: Apache Software License",
     "Intended Audience :: Science/Research",
     "Topic :: Scientific/Engineering",