Skip to content

Commit

Permalink
Use SDK version that matches PyFunc servers
Browse files Browse the repository at this point in the history
  • Loading branch information
Krithika Sundararajan committed Nov 29, 2023
1 parent 67b0583 commit 76d919f
Show file tree
Hide file tree
Showing 2 changed files with 2 additions and 2 deletions.
2 changes: 1 addition & 1 deletion python/batch-predictor/docker/app.Dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -25,6 +25,6 @@ RUN if [[ ! -z "$GOOGLE_APPLICATION_CREDENTIALS" ]]; then gcloud auth activate-s
RUN gsutil -m cp -r ${MODEL_URL} .
RUN /bin/bash -c ". activate ${CONDA_ENVIRONMENT} && \
sed -i 's/mlflow[><=]\{0,1\}.*$/mlflow==1.26.1/g' ${HOME}/model/conda.yaml && \
sed -i '/merlin-sdk/d' ${HOME}/model/conda.yaml && \
conda env update --name ${CONDA_ENVIRONMENT} --file ${HOME}/model/conda.yaml && \
pip install ${SDK_PATH} && \
python ${HOME}/merlin-spark-app/main.py --dry-run-model ${HOME}/model"
2 changes: 1 addition & 1 deletion python/pyfunc-server/docker/Dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -25,8 +25,8 @@ RUN if [ ! -z "$GOOGLE_APPLICATION_CREDENTIALS" ]; then gcloud auth activate-ser
RUN gsutil cp -r ${MODEL_URL} .
RUN /bin/bash -c ". activate merlin-model && \
sed -i 's/mlflow[><=]\{0,1\}.*$/mlflow==1.26.1/g' model/conda.yaml && \
sed -i '/merlin-sdk/d' model/conda.yaml && \
conda env update --name merlin-model --file model/conda.yaml && \
pip install ${SDK_PATH} && \
python -m pyfuncserver --model_dir model --dry_run"

CMD ["/bin/bash", "./run.sh"]

0 comments on commit 76d919f

Please sign in to comment.