From 8c02199e25499e8a8ae9e9363b23b7ca4cbfa24b Mon Sep 17 00:00:00 2001
From: Colin Gaffney <cpgaffney@google.com>
Date: Fri, 15 Nov 2024 11:38:08 -0800
Subject: [PATCH] Add chex as an optional testing dependency to unbreak RTD.
 Unclear why this has not previously been a problem.

PiperOrigin-RevId: 696952367
---
 checkpoint/pyproject.toml | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/checkpoint/pyproject.toml b/checkpoint/pyproject.toml
index 68eb7dd2..96955295 100644
--- a/checkpoint/pyproject.toml
+++ b/checkpoint/pyproject.toml
@@ -50,5 +50,6 @@ testing = [
     'mock',
     'flax',
     'pytest',
-    'pytest-xdist'
+    'pytest-xdist',
+    'chex',
 ]