From 2f1383c1561979889ba37a5f5f63c12f77145a22 Mon Sep 17 00:00:00 2001
From: siege <siege@google.com>
Date: Tue, 7 Jan 2025 11:26:51 -0800
Subject: [PATCH] Temporarily fix the OSS tests.

- Pin hypothesis to an older version, as newer versions reveal too many miscalibrated tests.
- Forbid recent TF nightly packages, as they crash on import in Inference Gym.

PiperOrigin-RevId: 712983081
---
 testing/dependency_install_lib.sh | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/testing/dependency_install_lib.sh b/testing/dependency_install_lib.sh
index f6b9a2a65b..09679f8a8e 100644
--- a/testing/dependency_install_lib.sh
+++ b/testing/dependency_install_lib.sh
@@ -61,7 +61,7 @@ find_good_tf_nightly_version_str() {
   VERSION=$1
   curl -s "https://pypi.org/pypi/${VERSION}/json" \
     | python -c "$PYTHON_PARSE_PACKAGE_JSON" \
-        --bad_dates 20220727 20220809 20220811 20220902 20230105
+        --bad_dates 20250103 20250105 20250107
 }
 
 install_tensorflow() {
@@ -96,7 +96,7 @@ install_test_only_packages() {
     bayeux-ml \
     chex \
     flax \
-    hypothesis \
+    hypothesis==6.80.0 \
     jax \
     jaxlib \
     jaxtyping \