From 24b63a22df02d9ec050b12b2943c238957bf3b65 Mon Sep 17 00:00:00 2001
From: Uri Granta <uri.granta@secondmind.ai>
Date: Tue, 18 Jun 2024 10:11:41 +0100
Subject: [PATCH] setup

---
 setup.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/setup.py b/setup.py
index e708f16e..33ab453f 100644
--- a/setup.py
+++ b/setup.py
@@ -9,9 +9,9 @@
     "gpflow>=2.9.2",
     "numpy",
     "scipy",
-    "tensorflow>=2.5.0,<=2.16; platform_system!='Darwin' or platform_machine!='arm64'",
+    "tensorflow>=2.5.0,<=2.16.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.16; platform_system=='Darwin' and platform_machine=='arm64'",
+    "tensorflow-macos>=2.5.0,<=2.16.1; platform_system=='Darwin' and platform_machine=='arm64'",
     "tensorflow-probability>=0.13.0,<=0.24",
 ]