From ca004cca3082f8cd82b8480223d4320c3d120797 Mon Sep 17 00:00:00 2001
From: Ze Mao <zemao@google.com>
Date: Wed, 27 Nov 2024 16:08:08 -0800
Subject: [PATCH] chore(components): Update Dockerfile

Signed-off-by: Ze Mao <zemao@google.com>
PiperOrigin-RevId: 700828757
---
 components/google-cloud/Dockerfile | 11 ++---------
 1 file changed, 2 insertions(+), 9 deletions(-)

diff --git a/components/google-cloud/Dockerfile b/components/google-cloud/Dockerfile
index 595241ca81e..4728c5d8cfa 100644
--- a/components/google-cloud/Dockerfile
+++ b/components/google-cloud/Dockerfile
@@ -16,18 +16,11 @@
 FROM marketplace.gcr.io/google/ubuntu2404:latest
 RUN apt update && apt -y install git python3 \
   python3-pip \
-  python3-venv
+  python3-setuptools
 
 WORKDIR /root
 
-# Create a virtual environment
-RUN python3 -m venv venv
-
-# Activate the virtual environment
-ENV PATH="venv/bin:$PATH"
-
-# Python 3.12 removed distutils
-RUN pip3 install -U pip setuptools
+ENV PIP_BREAK_SYSTEM_PACKAGES=1
 
 # Required by gcp_launcher
 # Using google-cloud-aiplatform>=1.21.0 to avoid dataset creatation timeout