Drop support for 3.8. See https://jax.readthedocs.io/en/latest/change…

…log.html#jax-0-4-14.

PiperOrigin-RevId: 543418693

.github/workflows/ci.yml

@@ -15,7 +15,7 @@ jobs:
 
     strategy:
       matrix:
-        python-version: ["3.8", "3.9", "3.10"]
+        python-version: ["3.9", "3.10", "3.11"]
         os: [ubuntu-latest]
 
     steps:

setup.py

@@ -58,7 +58,7 @@ def _parse_requirements(path):
         os.path.join(_CURRENT_DIR, 'requirements', 'requirements-test.txt')),
     zip_safe=False,  # Required for full installation.
     include_package_data=True,
-    python_requires='>=3.8',
+    python_requires='>=3.9',
     classifiers=[
         'Development Status :: 5 - Production/Stable',
         'Environment :: Console',

test.sh

@@ -25,7 +25,7 @@ python --version
 
 # Install dependencies.
 pip install --upgrade pip setuptools wheel
-pip install flake8 pytest-xdist pytype pylint pylint-exit
+pip install flake8 pytest-xdist pylint pylint-exit
 pip install -r requirements/requirements.txt
 pip install -r requirements/requirements-test.txt
 
@@ -52,7 +52,13 @@ pip wheel --verbose --no-deps --no-clean dist/chex*.tar.gz
 pip install chex*.whl
 
 # Check types with pytype.
-pytype `find chex/_src/ -name "*py" | xargs` -k
+# Note: pytype does not support 3.11 as of 25.06.23
+# See https://github.com/google/pytype/issues/1308
+if [ `python -c 'import sys; print(sys.version_info.minor)'` -lt 11 ];
+then
+  pip install pytype
+  pytype `find chex/_src/ -name "*py" | xargs` -k
+fi;
 
 # Run tests using pytest.
 # Change directory to avoid importing the package from repo root.