Skip to content

Commit 082d086

Browse files
committed
Fix sklearnex: pin scikit-learn-intelex>=2025.10.1 for NumPy 2.4 compat
1 parent 53f0551 commit 082d086

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

Dockerfile.tmpl

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -30,8 +30,8 @@ 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"
3131
# b/493600019: Colab base image ships numba that does not support NumPy 2.4; upgrade to latest.
3232
RUN uv pip install --system --force-reinstall --no-cache numba
33-
# b/493600019: New Colab base image may drop scikit-learn-intelex; re-install to ensure sklearnex is available.
34-
RUN uv pip install --system --force-reinstall --no-cache scikit-learn-intelex
33+
# b/493600019: New Colab base image drops scikit-learn-intelex; install >=2025.10.1 which has NumPy 2.4 fixes.
34+
RUN uv pip install --system --no-cache "scikit-learn-intelex>=2025.10.1"
3535

3636
# Adding non-package dependencies:
3737
ADD clean-layer.sh /tmp/clean-layer.sh

0 commit comments

Comments
 (0)