diff --git a/Dockerfile.tmpl b/Dockerfile.tmpl index c4df4612..38c87e69 100644 --- a/Dockerfile.tmpl +++ b/Dockerfile.tmpl @@ -114,11 +114,9 @@ RUN pip uninstall -y pyarrow && \ mamba install -y cudf cuml && \ /tmp/clean-layer.sh -# TODO(neshdev): Resolve pandas dependency another way -RUN cat /opt/conda/lib/python3.10/site-packages/cudf/api/types.py \ - | sed 's/^is_extension_type/# is_extension_type/g' \ - | sed 's/^is_categorical/# is_categorical/g' \ - > /opt/conda/lib/python3.10/site-packages/cudf/api/types.py +# TODO: b/296444923 - Resolve pandas dependency another way +RUN sed -i 's/^is_extension_type/# is_extension_type/g' /opt/conda/lib/python3.10/site-packages/cudf/api/types.py \ + && sed -i 's/^is_categorical/# is_categorical/g' /opt/conda/lib/python3.10/site-packages/cudf/api/types.py {{ end }} # Install PyTorch