Add chex as an optional testing dependency to unbreak RTD. Unclear wh…

…y this has not previously been a problem. PiperOrigin-RevId: 696952367
google · Nov 15, 2024 · 8c02199 · 8c02199
1 parent 9aa1e58
commit 8c02199
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/checkpoint/pyproject.toml b/checkpoint/pyproject.toml
@@ -50,5 +50,6 @@ testing = [
     'mock',
     'flax',
     'pytest',
-    'pytest-xdist'
+    'pytest-xdist',
+    'chex',
 ]