From 5ed919d1e35429a855c4d019373c42a66dcb0538 Mon Sep 17 00:00:00 2001 From: bvolpato Date: Tue, 27 Oct 2026 01:14:43 -0400 Subject: [PATCH] [CI] Use Python 3.11 in the vLLM benchmark image --- .../ml/inference/test_resources/vllm.dockerfile | 17 ++++++++++------- 1 file changed, 10 insertions(+), 7 deletions(-) diff --git a/sdks/python/apache_beam/ml/inference/test_resources/vllm.dockerfile b/sdks/python/apache_beam/ml/inference/test_resources/vllm.dockerfile index f4022ae90160..b3203a17a246 100644 --- a/sdks/python/apache_beam/ml/inference/test_resources/vllm.dockerfile +++ b/sdks/python/apache_beam/ml/inference/test_resources/vllm.dockerfile @@ -28,25 +28,28 @@ RUN apt-get update && \ curl \ tzdata \ software-properties-common \ - python3.10-full \ - python3.10-distutils \ build-essential \ - python3.10-dev \ cython3 && \ + add-apt-repository -y ppa:deadsnakes/ppa && \ + apt-get update && \ + apt-get install -y --no-install-recommends \ + python3.11-full \ + python3.11-distutils \ + python3.11-dev && \ ln -fs /usr/share/zoneinfo/$TZ /etc/localtime && \ dpkg-reconfigure --frontend noninteractive tzdata && \ rm -rf /var/lib/apt/lists/* -# 2) Symlink python3 to 3.10 -RUN ln -sf /usr/bin/python3.10 /usr/bin/python3 && \ - ln -sf /usr/bin/python3.10 /usr/bin/python +# 2) Symlink python3 to 3.11 +RUN ln -sf /usr/bin/python3.11 /usr/bin/python3 && \ + ln -sf /usr/bin/python3.11 /usr/bin/python # 3) Install pip, setuptools & wheel RUN curl -sS https://bootstrap.pypa.io/get-pip.py | python3 && \ python3 -m pip install --upgrade pip setuptools wheel # 4) Copy the Beam SDK harness (for Dataflow workers) -COPY --from=gcr.io/apache-beam-testing/beam-sdk/beam_python3.10_sdk:latest \ +COPY --from=gcr.io/apache-beam-testing/beam-sdk/beam_python3.11_sdk:latest \ /opt/apache/beam /opt/apache/beam # 5) Make sure the harness is discovered first