Add CI for Python 3.10

This removes sphinxcontrib-bibtex as a dependency, which needs to be upgraded to
support Python 3.10 and was unused.

PiperOrigin-RevId: 503938136

.github/workflows/ci.yml

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

.readthedocs.yaml

@@ -9,7 +9,7 @@ sphinx:
   fail_on_warning: false
 
 python:
-  version: 3.7
+  version: 3.8
   install:
     - requirements: requirements/requirements-docs.txt
     - requirements: requirements/requirements.txt

chex/_src/dataclass_test.py

@@ -15,7 +15,6 @@
 """Tests for `dataclass.py`."""
 import copy
 import dataclasses
-import sys
 from typing import Any, Mapping, TypeVar, Generic
 
 from absl.testing import absltest
@@ -499,12 +498,6 @@ def _is_leaf(value) -> bool:
       ('not_mappable', False),
   )
   def test_generic_dataclass(self, mappable):
-    # Running under Python 3.6 results in error "TypeError: Cannot inherit from
-    # plain Generic", I'm speculating that this is a bug in cpython since
-    # subsequent versions work
-    if sys.version_info < (3, 7):
-      self.skipTest('Skip test on Python version < 3.7')
-
     T = TypeVar('T')
 
     @chex_dataclass(mappable_dataclass=mappable)

docs/conf.py

@@ -94,7 +94,6 @@ def _recursive_add_annotations_import():
     'sphinx.ext.intersphinx',
     'sphinx.ext.linkcode',
     'sphinx.ext.napoleon',
-    'sphinxcontrib.bibtex',
     'sphinxcontrib.katex',
     'sphinx_autodoc_typehints',
     'sphinx_rtd_theme',

requirements/requirements-docs.txt

@@ -1,7 +1,6 @@
 sphinx==4.5.0
 sphinx_rtd_theme==1.0.0
 sphinxcontrib-katex==0.8.6
-sphinxcontrib-bibtex==1.0.0
 sphinx-autodoc-typehints==1.11.1
 IPython==7.16.3
 ipykernel==5.3.4

requirements/requirements.txt

@@ -1,5 +1,4 @@
 absl-py>=0.9.0
-dataclasses>=0.7; python_version>="3.6" and python_version<"3.7"
 typing_extensions>=4.2.0; python_version<"3.11"
 dm-tree>=0.1.5
 jax>=0.1.55

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.7',
+    python_requires='>=3.8',
     classifiers=[
         'Development Status :: 5 - Production/Stable',
         'Environment :: Console',