Pin scikit-learn to 0.23.2

The new 0.24.0 version has been released on Dec 23 but isn't compatible with several of our packages including hep-ml. BUG=176817038
st01cs · Jan 5, 2021 · dd0914d · dd0914d
1 parent 915befa
commit dd0914d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/Dockerfile b/Dockerfile
@@ -116,9 +116,9 @@ RUN pip install ibis-framework && \
  pip install gluoncv && \
  /tmp/clean-layer.sh
 
-# scikit-learn dependencies
 RUN pip install scipy && \
- pip install scikit-learn && \
+ # b/176817038 avoid upgrade to 0.24 which is causing issues with hep-ml package.
+ pip install scikit-learn==0.23.2 && \
  # HDF5 support
  pip install h5py && \
  pip install biopython && \