Skip to content

Commit c8d8be7

Browse files
committed
Fix sklearnex: force-reinstall scikit-learn-intelex after base image upgrade
1 parent 627cbdb commit c8d8be7

1 file changed

Lines changed: 2 additions & 0 deletions

File tree

Dockerfile.tmpl

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -28,6 +28,8 @@ RUN uv pip install --no-build-isolation --no-cache --system "git+https://github.
2828
# b/468367647: Unpin protobuf, version greater than v5.29.5 causes issues with numerous packages
2929
RUN uv pip install --system --force-reinstall --no-cache --no-deps torchtune
3030
RUN uv pip install --system --force-reinstall --no-cache "protobuf==5.29.5"
31+
# b/493600019: New Colab base image may drop scikit-learn-intelex; re-install to ensure sklearnex is available.
32+
RUN uv pip install --system --force-reinstall --no-cache scikit-learn-intelex
3133

3234
# Adding non-package dependencies:
3335
ADD clean-layer.sh /tmp/clean-layer.sh

0 commit comments

Comments
 (0)