From 1a827192b6a7286777f7b84ad2550432a966c0a7 Mon Sep 17 00:00:00 2001
From: Uri Granta <uri.granta@secondmind.ai>
Date: Fri, 6 Sep 2024 15:28:59 +0100
Subject: [PATCH] Add upper bound to TF in setup

---
 setup.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/setup.py b/setup.py
index e164ff96a..47aab3fca 100644
--- a/setup.py
+++ b/setup.py
@@ -43,9 +43,9 @@
         "gpflow>=2.9.2",
         "gpflux>=0.4.4",
         "numpy",
-        "tensorflow>=2.5; platform_system!='Darwin' or platform_machine!='arm64'",
-        "tensorflow-macos>=2.5; platform_system=='Darwin' and platform_machine=='arm64'",
-        "tensorflow-probability>=0.13",
+        "tensorflow>=2.5,<2.17; platform_system!='Darwin' or platform_machine!='arm64'",
+        "tensorflow-macos>=2.5,<2.17; platform_system=='Darwin' and platform_machine=='arm64'",
+        "tensorflow-probability>=0.13,<0.25",
         "greenlet>=1.1.0",
     ],
     extras_require={