From 74cc4f6fc362baab4ba2ac4d8e5575b90820ae8a Mon Sep 17 00:00:00 2001
From: Codex <codex@openai.com>
Date: Sat, 6 Jun 2026 23:38:41 -0400
Subject: [PATCH] Fix CI dependency installation

---
 .github/workflows/ci.yml | 4 ++--
 MANIFEST.in              | 1 +
 README.md                | 2 +-
 requirements-test.txt    | 1 -
 requirements-tf.txt      | 3 +--
 setup.py                 | 3 +--
 6 files changed, 6 insertions(+), 8 deletions(-)

diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml
index 630e704b..e6451219 100644
--- a/.github/workflows/ci.yml
+++ b/.github/workflows/ci.yml
@@ -17,9 +17,9 @@ jobs:
         python-version: [3.9, '3.10', '3.11']
         os: [ubuntu-latest]
     steps:
-    - uses: actions/checkout@v2
+    - uses: actions/checkout@v4
     - name: Set up Python ${{ matrix.python-version }}
-      uses: actions/setup-python@v1
+      uses: actions/setup-python@v5
       with:
         python-version: ${{ matrix.python-version }}
     - name: Install dependencies
diff --git a/MANIFEST.in b/MANIFEST.in
index 04f196ac..6c36e39e 100644
--- a/MANIFEST.in
+++ b/MANIFEST.in
@@ -1,2 +1,3 @@
 include README.md
 include LICENSE
+include requirements*.txt
diff --git a/README.md b/README.md
index ebd3247e..e77f05c6 100644
--- a/README.md
+++ b/README.md
@@ -47,7 +47,7 @@ notebook attached to a GPU or TPU.
 To get started install TensorFlow 2.0 and Sonnet 2:
 
 ```shell
-$ pip install tensorflow tensorflow-probability
+$ pip install tensorflow
 $ pip install dm-sonnet
 ```
 
diff --git a/requirements-test.txt b/requirements-test.txt
index be42e585..6eb61780 100644
--- a/requirements-test.txt
+++ b/requirements-test.txt
@@ -1,3 +1,2 @@
-mock>=3.0.5
 tensorflow-datasets>1,<4
 docutils
diff --git a/requirements-tf.txt b/requirements-tf.txt
index 5ca90d45..5450c277 100644
--- a/requirements-tf.txt
+++ b/requirements-tf.txt
@@ -1,2 +1 @@
-tensorflow==2.12.0rc0
-tensorflow-probability==0.12.2
+tensorflow==2.12.1
diff --git a/setup.py b/setup.py
index e1e3ddd1..b8122d75 100644
--- a/setup.py
+++ b/setup.py
@@ -41,8 +41,7 @@ def _parse_requirements(requirements_txt_path):
     packages=find_namespace_packages(exclude=['*_test.py']),
     install_requires=_parse_requirements('requirements.txt'),
     extras_require=EXTRA_PACKAGES,
-    tests_require=_parse_requirements('requirements-test.txt'),
-    requires_python='>=3.6',
+    python_requires='>=3.9,<3.12',
     include_package_data=True,
     zip_safe=False,
     # PyPI package information.