Pin the NumPy version with v1.23.5 in the mxnet-mnist and the skopt s…

…uggestion service image Signed-off-by: Yuki Iwai <[email protected]>
kubeflow · Dec 23, 2022 · 3c696f8 · 3c696f8
1 parent db72ce1
commit 3c696f8
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 2 deletions.
diff --git a/cmd/suggestion/skopt/v1beta1/requirements.txt b/cmd/suggestion/skopt/v1beta1/requirements.txt
@@ -1,6 +1,9 @@
 grpcio==1.41.1
 cloudpickle==0.5.6
-numpy>=1.20.0
+# This is a workaround to avoid the following error.
+# AttributeError: module 'numpy' has no attribute 'bool'
+# See more: https://github.com/numpy/numpy/pull/22607
+numpy==1.23.5
 scikit-learn>=0.24.0
 scipy>=1.5.4
 forestci==0.3

diff --git a/examples/v1beta1/trial-images/mxnet-mnist/Dockerfile b/examples/v1beta1/trial-images/mxnet-mnist/Dockerfile
@@ -20,7 +20,7 @@ RUN apt-get -y update \
 RUN if [ "${TARGETARCH}" = "arm64" ]; then \
       /opt/mxnet-mnist/install-arm-performance-libraries.sh; \
     fi
-RUN pip install mxnet==1.9.1
+RUN pip install -r requirements.txt
 RUN chgrp -R 0 /opt/mxnet-mnist \
   && chmod -R g+rwX /opt/mxnet-mnist
 

diff --git a/examples/v1beta1/trial-images/mxnet-mnist/requirements.txt b/examples/v1beta1/trial-images/mxnet-mnist/requirements.txt
@@ -0,0 +1,5 @@
+mxnet==1.9.1
+# This is a workaround to avoid the following error.
+# AttributeError: module 'numpy' has no attribute 'bool'
+# See more: https://github.com/numpy/numpy/pull/22607
+numpy==1.23.5