Pin the tensorflow version lower than 2.16 (#215)

TileDB-Inc · Aug 26, 2024 · 3b42e5b · 3b42e5b
1 parent effa1c1
commit 3b42e5b
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git 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
@@ -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"]