Add upper bound to TF in setup

secondmind-labs · Sep 6, 2024 · 1a82719 · 1a82719
1 parent 9893de8
commit 1a82719
Showing 1 changed file with 3 additions and 3 deletions.
diff --git 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={