Merge pull request #159 from aws/pyarrow-fix

PyArrow/Numpy Version Bump Error Fix

Author: amzn-choeric

docker/1.0-1/base/Dockerfile.cpu

@@ -8,8 +8,9 @@ ARG CONDA_CHECKSUM=3190da6626f86eee8abf1b2fd7a5af492994eb2667357ee4243975cdbb175
 ARG CONDA_PY_VERSION=38
 ARG CONDA_PKG_VERSION=4.13.0
 ARG PYTHON_VERSION=3.8.13
-ARG PYARROW_VERSION=1.0.0
-ARG MLIO_VERSION=0.7.0
+ARG PYARROW_VERSION=10.0.1
+ARG MLIO_VERSION=v0.8.0
+ARG NUMPY_VERSION=1.24.1
 
 # Install python and other scikit-learn runtime dependencies
 # Dependency list from http://scikit-learn.org/stable/developers/advanced_installation.html#installing-build-dependencies
@@ -81,9 +82,10 @@ RUN echo "conda ${CONDA_PKG_VERSION}" >> /miniconda3/conda-meta/pinned && \
     conda install -c conda-forge python=${PYTHON_VERSION} && \
     conda install conda=${CONDA_PKG_VERSION} && \
     conda update -y conda && \
+    conda install -c conda-forge numpy=${NUMPY_VERSION} && \
     conda install -c conda-forge pyarrow=${PYARROW_VERSION} && \
     cd /tmp && \
-    git clone --branch v${MLIO_VERSION} https://github.com/awslabs/ml-io.git mlio && \
+    git clone --branch ${MLIO_VERSION} https://github.com/awslabs/ml-io.git mlio && \
     cd mlio && \
     build-tools/build-dependency build/third-party all && \
     mkdir -p build/release && \

requirements.txt

@@ -6,7 +6,6 @@ itsdangerous==2.0.1
 gunicorn==20.0.4
 model-archiver==1.0.3
 multi-model-server==1.1.1
-numpy==1.22
 pandas==1.1.3
 protobuf==3.20.2
 psutil==5.7.2