Fix Numpy & h5py bug in 1.7.0 (opensearch-project#562)

Signed-off-by: Ian Hoang <[email protected]> Co-authored-by: Ian Hoang <[email protected]>
finnroblin · Jul 19, 2024 · 351b1b5 · 351b1b5
1 parent 0d9a565
commit 351b1b5
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/docker/Dockerfile b/docker/Dockerfile
@@ -15,7 +15,7 @@ RUN apt-get -y update && \
 RUN groupadd --gid 1000 opensearch-benchmark && \
     useradd -d /opensearch-benchmark -m -k /dev/null -g 1000 -N -u 1000 -l -s /bin/bash benchmark
 
-RUN python3 -m pip install h5py==3.10.0; if [ -z "$VERSION" ] ; then python3 -m pip install opensearch-benchmark ; else python3 -m pip install opensearch-benchmark==$VERSION ; fi
+RUN if [ -z "$VERSION" ] ; then python3 -m pip install opensearch-benchmark ; else python3 -m pip install opensearch-benchmark==$VERSION ; fi
 
 WORKDIR /opensearch-benchmark
 

diff --git a/setup.py b/setup.py
@@ -106,7 +106,7 @@ def str_from_file(name):
     "h5py>=3.10.0",
     # License: BSD
     # Required for knnvector workload
-    "numpy>=1.24.2",
+    "numpy>=1.24.2,<=1.26.4",
 ]
 
 tests_require = [