Pin the numpy version with v1.23.5 in the mxnet-mnist image

Signed-off-by: Yuki Iwai <[email protected]>
kubeflow · Dec 23, 2022 · 4b364ad · 4b364ad
1 parent db72ce1
commit 4b364ad
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
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