From 17a67d97caabb08c9b5ce090a82e7e3948788c05 Mon Sep 17 00:00:00 2001 From: sidiyer20 Date: Thu, 26 May 2022 15:17:02 +0530 Subject: [PATCH] adding env variable for protobuf error --- mlflow/models/docker_utils.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/mlflow/models/docker_utils.py b/mlflow/models/docker_utils.py index d1a5f7ff5a42c..93ff8db550b33 100644 --- 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 \