Skip to content

Commit d4f9531

Browse files
author
aria
authored
Force pyfunc server to use mlflow 1.6.0 (#172)
1 parent bbbcf02 commit d4f9531

File tree

2 files changed

+2
-0
lines changed

2 files changed

+2
-0
lines changed

python/pyfunc-server/Dockerfile

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -28,6 +28,7 @@ RUN conda env create -f ./environment.yaml && \
2828

2929
RUN gsutil cp -r ${MODEL_URL} .
3030
RUN /bin/bash -c ". activate merlin-model && \
31+
sed -i 's/mlflow$/mlflow==1.6.0/' /model/conda.yaml && \
3132
conda env update --name merlin-model --file /model/conda.yaml && \
3233
python -m pyfuncserver --model_dir /model --dry_run"
3334

python/pyfunc-server/local.Dockerfile

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -24,6 +24,7 @@ RUN conda env create -f ./environment.yaml && \
2424
rm -rf /root/.cache
2525

2626
RUN /bin/bash -c ". activate merlin-model && \
27+
sed -i 's/mlflow$/mlflow==1.6.0/' /model/conda.yaml && \
2728
conda env update --name merlin-model --file /model/conda.yaml && \
2829
python -m pyfuncserver --model_dir /model --dry_run"
2930

0 commit comments

Comments
 (0)