diff --git a/python/batch-predictor/docker/app.Dockerfile b/python/batch-predictor/docker/app.Dockerfile index a23205ff8..80f95eee2 100644 --- a/python/batch-predictor/docker/app.Dockerfile +++ b/python/batch-predictor/docker/app.Dockerfile @@ -31,7 +31,7 @@ COPY --chown=${UID}:${GID} batch-predictor ${HOME}/merlin-spark-app COPY --chown=${UID}:${GID} sdk ${HOME}/sdk ENV SDK_PATH=${HOME}/sdk -RUN /bin/bash -c ". activate merlin-model && pip install -r ${HOME}/merlin-spark-app/requirements.txt" +RUN /bin/bash -c ". activate merlin-model && pip uninstall -y merlin-sdk && pip install -r ${HOME}/merlin-spark-app/requirements.txt" # Download and dry-run user model artifacts and code ARG MODEL_ARTIFACTS_URL diff --git a/python/pyfunc-server/docker/Dockerfile b/python/pyfunc-server/docker/Dockerfile index 8671797ac..71c5fbcc4 100644 --- a/python/pyfunc-server/docker/Dockerfile +++ b/python/pyfunc-server/docker/Dockerfile @@ -32,7 +32,7 @@ COPY sdk /sdk ENV SDK_PATH=/sdk WORKDIR /pyfunc-server -RUN /bin/bash -c ". activate merlin-model && pip install -r /pyfunc-server/requirements.txt" +RUN /bin/bash -c ". activate merlin-model && pip uninstall -y merlin-sdk && pip install -r /pyfunc-server/requirements.txt" # Download and dry-run user model artifacts and code ARG MODEL_ARTIFACTS_URL