Do not pre-install beam in tensorrt container (#30552)

apache · Mar 7, 2024 · a18c608 · a18c608
1 parent 9b71fad
commit a18c608
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/sdks/python/test-suites/containers/tensorrt_runinference/tensor_rt.dockerfile b/sdks/python/test-suites/containers/tensorrt_runinference/tensor_rt.dockerfile
@@ -22,8 +22,7 @@ ENV PATH="/usr/src/tensorrt/bin:${PATH}"
 
 WORKDIR /workspace
 
-RUN pip install --no-cache-dir apache-beam[gcp]==2.48.0
-COPY --from=apache/beam_python3.8_sdk:2.48.0 /opt/apache/beam /opt/apache/beam
+COPY --from=apache/beam_python3.8_sdk:latest /opt/apache/beam /opt/apache/beam
 
 RUN pip install --upgrade pip \
     && pip install torch>=1.7.1 \