Skip to content

Commit a5d8e6c

Browse files
committed
fix uninstalls
1 parent 8de0804 commit a5d8e6c

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

Dockerfile.tmpl

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -72,7 +72,7 @@ RUN uv pip install --system boto3==1.26.100 && \
7272

7373
{{ if eq .Accelerator "gpu" }}
7474
# b/200968891 Keeps horovod once torch is upgraded.
75-
RUN uv pip uninstall --system -y horovod && \
75+
RUN uv pip uninstall --system horovod && \
7676
/tmp/clean-layer.sh
7777
{{ end }}
7878

@@ -113,7 +113,7 @@ RUN conda config --add channels nvidia && \
113113
# b/232247930: uninstall pyarrow to avoid double installation with the GPU specific version.
114114
# b/341938540: unistall grpc-cpp to allow >=v24.4 cudf and cuml to be installed.
115115
{{ if eq .Accelerator "gpu" }}
116-
RUN uv pip uninstall --system -y pyarrow && \
116+
RUN uv pip uninstall --system pyarrow && \
117117
mamba remove -y --force grpc-cpp && \
118118
mamba install -y -c conda-forge spacy cudf>=24.4 cuml>=24.4 cupy cuda-version=$CUDA_MAJOR_VERSION.$CUDA_MINOR_VERSION && \
119119
/tmp/clean-layer.sh
@@ -385,7 +385,7 @@ RUN uv pip install --system cython \
385385
RUN uv pip install --system annoy \
386386
category_encoders && \
387387
# b/183041606#comment5: the Kaggle data proxy doesn't support these APIs. If the library is missing, it falls back to using a regular BigQuery query to fetch data.
388-
pip uninstall -y google-cloud-bigquery-storage && \
388+
uv pip uninstall --system google-cloud-bigquery-storage && \
389389
# google-cloud-automl 2.0.0 introduced incompatible API changes, need to pin to 1.0.1
390390
# After launch this should be installed from pip
391391
uv pip install --system git+https://github.com/googleapis/python-aiplatform.git@mb-release \

0 commit comments

Comments
 (0)