* Migrate to pyproject.toml

* Drop support for python 3.10

PiperOrigin-RevId: 801826037

Author: hbq1

.github/workflows/ci.yml

@@ -23,13 +23,10 @@ jobs:
 
     strategy:
       matrix:
-        python-version: ["3.10", "3.11", "3.12", "3.13"]
+        python-version: ["3.11", "3.12", "3.13"]
         os: [ubuntu-latest]
         jax-version: ["newest"]
         include:
-          - python-version: "3.10"
-            os: "ubuntu-latest"
-            jax-version: "0.4.27"  # Keep this in sync with version in requirements.txt
           - python-version: "3.13"
             os: "ubuntu-latest"
             jax-version: "nightly"

.github/workflows/pypi-publish.yml

@@ -22,7 +22,7 @@ jobs:
     - name: Check consistency between the package version and release tag
       run: |
         RELEASE_VER=${GITHUB_REF#refs/*/}
-        PACKAGE_VER="v`python setup.py --version`"
+        PACKAGE_VER="v`python -c 'import chex; print(chex.__version__)'`"
         if [ $RELEASE_VER != $PACKAGE_VER ]
         then
           echo "package ver. ($PACKAGE_VER) != release ver. ($RELEASE_VER)"; exit 1

.gitignore

@@ -5,7 +5,14 @@
 build/
 dist/
 venv/
+_testing/
+
+# Building the documentation
+docs/_autosummary
 docs/_build/
+docs/_collections
+docs/modules/generated
+docs/sg_execution_times.rst
 
 # Mac OS
 .DS_Store

.readthedocs.yaml

@@ -15,7 +15,11 @@ sphinx:
 
 python:
   install:
-    - requirements: requirements/requirements-docs.txt
-    - requirements: requirements/requirements.txt
-    - method: setuptools
+    # Equivalent to 'pip install .'
+    - method: pip
       path: .
+    # Equivalent to 'pip install .[docs]'
+    - method: pip
+      path: .
+      extra_requirements:
+        - docs

MANIFEST.in

@@ -0,0 +1,126 @@
+[build-system]
+requires = ["flit_core >=3.2,<4"]
+build-backend = "flit_core.buildapi"
+
+[project]
+name = "chex"
+dynamic = ["version"]
+description = "Chex: Testing made fun, in JAX!"
+readme = "README.md"
+license = { file = "LICENSE" }
+requires-python = ">=3.11"
+authors = [
+    {name = "Google DeepMind", email = "chex-dev@google.com"},
+]
+keywords = [
+    "jax",
+    "testing",
+    "debugging",
+    "python",
+    "machine learning",
+]
+classifiers = [
+    "Environment :: Console",
+    "Programming Language :: Python",
+    "Intended Audience :: Developers",
+    "Operating System :: OS Independent",
+    "Programming Language :: Python :: 3",
+    "Intended Audience :: Science/Research",
+    "Development Status :: 4 - Beta",
+    "License :: OSI Approved :: Apache Software License",
+    "Topic :: Scientific/Engineering :: Artificial Intelligence",
+    "Topic :: Software Development :: Libraries :: Python Modules",
+]
+dependencies = [
+    "absl-py>=2.3.1",
+    "typing_extensions>=4.15.0",
+    "jax>=0.7.0",
+    "jaxlib>=0.7.0",
+    "numpy>=1.24.1",
+    "toolz>=1.0.0",
+]
+
+[project.urls]
+homepage = "https://github.com/google-deepmind/chex"
+repository = "https://github.com/google-deepmind/chex"
+documentation = "https://chex.readthedocs.io/"
+
+[project.optional-dependencies]
+test = [
+    "cloudpickle==3.1.0",
+    "dm-tree>=0.1.9",
+]
+
+docs = [
+    "sphinx>=6.0.0",
+    "sphinx-book-theme>=1.0.1",
+    "sphinxcontrib-katex",
+]
+
+[tool.setuptools.packages.find]
+include = ["README.md", "LICENSE"]
+exclude = ["*_test.py"]
+
+[tool.ruff]
+line-length = 80
+
+[tool.ruff.lint]
+select = [
+    "F",
+    "E",
+    "W291",  # whitespace at the end of the line
+    "B023",  # pylint's cell-var-over-loop, closures capturing variables in loop
+]
+ignore = [
+    "E731",  # lambdas are allowed
+    "F401",  # allow unused imports
+    "E402",  # allow modules not at top of file
+    "E741",  # allow "l" as a variable name
+    "E703",  # allow semicolons (for jupyter notebooks)
+]
+
+[tool.pylint.messages_control]
+disable = [
+    "bad-indentation",
+    "unknown-option-value",
+    "invalid-name",
+    "missing-function-docstring",
+    "missing-class-docstring",
+    "missing-module-docstring",
+    "no-member",
+    "too-many-locals",
+    "too-many-positional-arguments",
+    "no-else-return",
+    "line-too-long",
+    "too-many-arguments",
+    "no-value-for-parameter",
+    "duplicate-code",
+    "unused-argument",
+    "too-few-public-methods",
+    "wrong-import-order",
+    "unused-import",
+    "wrong-import-position",
+    "unnecessary-lambda-assignment",
+    "too-many-lines",
+    "too-many-statements",
+    "deprecated-class",
+    "redefined-builtin",
+    "used-before-assignment",
+    "undefined-variable",
+    "protected-access",
+    "not-callable",
+    "redefined-outer-name",
+    "too-many-instance-attributes",
+    "missing-final-newline",
+    "too-many-public-methods",
+    "import-error",
+]
+
+# We include pyink to allow external contributors to optionally, but easily,
+# pass google internal formatting checks.
+[tool.pyink]
+pyink = true  # false would mean black is used directly without pyink features
+pyink-use-majority-quotes = true
+pyink-indentation = 2
+line-length = 80
+include = '\.pyi?$'