Merge pull request #3 from sidiyer20/master

adding env variable for protobuf error
Zomato · May 26, 2022 · de5ae42 · de5ae42
2 parents 1b2793b + 17a67d9
commit de5ae42
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/mlflow/models/docker_utils.py b/mlflow/models/docker_utils.py
@@ -14,7 +14,8 @@
 _DOCKERFILE_TEMPLATE = """
 # Build an image that can serve mlflow models.
 FROM ubuntu:18.04
-
+ARG PROTOCOL_BUFFERS_PYTHON_IMPLEMENTATION=python
+ENV PROTOCOL_BUFFERS_PYTHON_IMPLEMENTATION=python
 RUN apt-get -y update && apt-get install -y --no-install-recommends \
          wget \
          curl \