Force pyfunc server to use mlflow 1.6.0 (#172)

caraml-dev · Aug 27, 2021 · d4f9531 · d4f9531
1 parent bbbcf02
commit d4f9531
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/python/pyfunc-server/Dockerfile b/python/pyfunc-server/Dockerfile
@@ -28,6 +28,7 @@ RUN conda env create -f ./environment.yaml && \
 
 RUN gsutil cp -r ${MODEL_URL} .
 RUN /bin/bash -c ". activate merlin-model && \
+    sed -i 's/mlflow$/mlflow==1.6.0/' /model/conda.yaml && \
     conda env update --name merlin-model --file /model/conda.yaml && \
     python -m pyfuncserver --model_dir /model --dry_run"
 

diff --git a/python/pyfunc-server/local.Dockerfile b/python/pyfunc-server/local.Dockerfile
@@ -24,6 +24,7 @@ RUN conda env create -f ./environment.yaml && \
     rm -rf /root/.cache
 
 RUN /bin/bash -c ". activate merlin-model && \
+    sed -i 's/mlflow$/mlflow==1.6.0/' /model/conda.yaml && \
     conda env update --name merlin-model --file /model/conda.yaml && \
     python -m pyfuncserver --model_dir /model --dry_run"