Skip to content

Commit 6946ddd

Browse files
authored
Merge pull request #929 from Kaggle/fix-hep-ml
Pin scikit-learn to 0.23.2
2 parents 915befa + dd0914d commit 6946ddd

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

Dockerfile

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -116,9 +116,9 @@ RUN pip install ibis-framework && \
116116
pip install gluoncv && \
117117
/tmp/clean-layer.sh
118118

119-
# scikit-learn dependencies
120119
RUN pip install scipy && \
121-
pip install scikit-learn && \
120+
# b/176817038 avoid upgrade to 0.24 which is causing issues with hep-ml package.
121+
pip install scikit-learn==0.23.2 && \
122122
# HDF5 support
123123
pip install h5py && \
124124
pip install biopython && \

0 commit comments

Comments
 (0)