From 3b42e5b6944c3562edbad0493680d35af263200f Mon Sep 17 00:00:00 2001
From: Konstantinos Tsitsimpikos <ktsitsi@users.noreply.github.com>
Date: Mon, 26 Aug 2024 16:48:25 +0300
Subject: [PATCH] Pin the tensorflow version lower than 2.16 (#215)

---
 .github/workflows/ci.yml | 2 +-
 setup.py                 | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml
index bf499c4f..082c0ca9 100644
--- a/.github/workflows/ci.yml
+++ b/.github/workflows/ci.yml
@@ -52,7 +52,7 @@ jobs:
 
     - name: Run notebook examples
       run: |
-        pip install pytest-xdist nbmake matplotlib idx2numpy
+        pip install pytest-xdist nbmake matplotlib idx2numpy "numpy<2"
         pytest --disable-warnings --nbmake examples/{models,readers}
         # Run tiledb-cloud in parallel
         if [[ "${{ secrets.TILEDB_API_TOKEN }}" != "" ]]; then
diff --git a/setup.py b/setup.py
index f5db4639..43e89282 100644
--- a/setup.py
+++ b/setup.py
@@ -1,6 +1,6 @@
 import setuptools
 
-tensorflow = ["tensorflow>=2.6"]
+tensorflow = ["tensorflow>=2.6, <2.16.0"]
 pytorch = ["torch>=1.11", "torchdata"]
 sklearn = ["scikit-learn>=1.0"]
 cloud = ["tiledb-cloud"]