From c20ce90bc81ee106600e13868c559569eaa21189 Mon Sep 17 00:00:00 2001
From: Sanket Kamthe <15108821+sanket-kamthe@users.noreply.github.com>
Date: Wed, 10 Aug 2022 16:06:29 +0100
Subject: [PATCH 1/2] Update setup.py

Allow tf 2.9.1
---
 setup.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/setup.py b/setup.py
index eb130d9d..a60e0181 100644
--- a/setup.py
+++ b/setup.py
@@ -9,7 +9,7 @@
     "gpflow>=2.1",
     "numpy",
     "scipy",
-    "tensorflow>=2.5.0,<2.9.0; platform_system!='Darwin' or platform_machine!='arm64'",
+    "tensorflow>=2.5.0,<=2.9.1; platform_system!='Darwin' or platform_machine!='arm64'",
     # NOTE: Support of Apple Silicon MacOS platforms is in an experimental mode
     "tensorflow-macos>=2.5.0,<2.9.0; platform_system=='Darwin' and platform_machine=='arm64'",
     # NOTE: once we require tensorflow-probability>=0.12, we can remove our custom deepcopy handling

From 0bb117dc1e0e54e79adb3099d16d28bcad04b459 Mon Sep 17 00:00:00 2001
From: Sanket Kamthe <15108821+sanket-kamthe@users.noreply.github.com>
Date: Wed, 10 Aug 2022 17:01:37 +0000
Subject: [PATCH 2/2] Add support for TF29

---
 .github/workflows/quality-check.yaml | 5 ++++-
 setup.py                             | 4 ++--
 2 files changed, 6 insertions(+), 3 deletions(-)

diff --git a/.github/workflows/quality-check.yaml b/.github/workflows/quality-check.yaml
index 98a5a110..40eb29c2 100644
--- a/.github/workflows/quality-check.yaml
+++ b/.github/workflows/quality-check.yaml
@@ -22,7 +22,7 @@ jobs:
     strategy:
       matrix:
           python-version: ["3.7", "3.8", "3.9", "3.10"]
-          tensorflow: ["~=2.5.0", "~=2.6.0", "~=2.7.0", "~=2.8.0"]
+          tensorflow: ["~=2.5.0", "~=2.6.0", "~=2.7.0", "~=2.8.0", "~=2.9.1"]
           include:
             - tensorflow: "~=2.5.0"
               keras: "~=2.6.0"
@@ -36,6 +36,9 @@ jobs:
             - tensorflow: "~=2.8.0"
               keras: "~=2.8.0"
               tensorflow-probability: "~=0.16.0"
+            - tensorflow: "~=2.9.1"
+              keras: "~=2.9.0"
+              tensorflow-probability: "~=0.17.0"
           exclude:
             # These older versions of TensorFlow don't work with Python 3.10:
             - python-version: "3.10"
diff --git a/setup.py b/setup.py
index a60e0181..3a60a89d 100644
--- a/setup.py
+++ b/setup.py
@@ -9,9 +9,9 @@
     "gpflow>=2.1",
     "numpy",
     "scipy",
-    "tensorflow>=2.5.0,<=2.9.1; platform_system!='Darwin' or platform_machine!='arm64'",
+    "tensorflow>=2.5.0,<2.10.0; platform_system!='Darwin' or platform_machine!='arm64'",
     # NOTE: Support of Apple Silicon MacOS platforms is in an experimental mode
-    "tensorflow-macos>=2.5.0,<2.9.0; platform_system=='Darwin' and platform_machine=='arm64'",
+    "tensorflow-macos>=2.5.0,<2.10.0; platform_system=='Darwin' and platform_machine=='arm64'",
     # NOTE: once we require tensorflow-probability>=0.12, we can remove our custom deepcopy handling
     "tensorflow-probability>=0.13.0,<0.17.0",
     "protobuf~=3.19.0"