diff --git a/tfx/tools/docker/Dockerfile b/tfx/tools/docker/Dockerfile index 9fa9938175..3ae4f936a8 100644 --- a/tfx/tools/docker/Dockerfile +++ b/tfx/tools/docker/Dockerfile @@ -26,6 +26,7 @@ WORKDIR ${TFX_DIR} # Specify what version of dependent libraries will be used. See dependencies.py. ARG TFX_DEPENDENCY_SELECTOR ENV TFX_DEPENDENCY_SELECTOR=${TFX_DEPENDENCY_SELECTOR} +ENV USE_BAZEL_VERSION=6.5.0 RUN python -m pip install --upgrade pip wheel setuptools tomli