fix(components): Set google-cloud-aiplatform>=1.21.0 to ultimately av…

…oid dataset creation 900s timeout and remove the workaround. Reference: googleapis/python-aiplatform#1870 (comment) PiperOrigin-RevId: 522122033
kubeflow · Apr 5, 2023 · 5de4d78 · 5de4d78
1 parent 4a9bc2b
commit 5de4d78
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/components/google-cloud/google_cloud_pipeline_components/container/Dockerfile b/components/google-cloud/google_cloud_pipeline_components/container/Dockerfile
@@ -21,17 +21,15 @@ WORKDIR /root
 RUN pip3 install --upgrade pip
 
 # Required by gcp_launcher
-RUN pip3 install -U google-cloud-aiplatform
+# Using google-cloud-aiplatform>=1.21.0 to avoid dataset creatation timeout
+RUN pip3 install -U "google-cloud-aiplatform>=1.21.0"
 RUN pip3 install -U google-cloud-storage
 RUN pip3 install -U google-api-python-client
 
 # Required by dataflow_launcher
 # b/238481913: Pinning the version of apache_beam to below 2.34 for now
 RUN pip3 install -U "apache_beam[gcp]<2.34.0"
 
-# b/267359259:  Pinning the version of google-api-core to below 1.34.0 for now
-RUN pip3 install -U "google-api-core<1.34.0"
-
 # Required for sklearn/train_test_split_jsonl
 RUN pip3 install -U "fsspec>=0.7.4" "gcsfs>=0.6.0" "pandas<=1.3.5" "scikit-learn<=1.0.2"