Skip to content

Commit 4e6bec8

Browse files
authored
Re-pin learntools related packages. (#1448)
These were unpinned during the base image migration but caused learntools to fail. http://b/365782129
1 parent 66bac48 commit 4e6bec8

File tree

2 files changed

+19
-7
lines changed

2 files changed

+19
-7
lines changed

Dockerfile.tmpl

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -21,10 +21,11 @@ ENV PATH="~/.local/bin:${PATH}"
2121
# b/183041606#comment5: the Kaggle data proxy doesn't support these APIs. If the library is missing, it falls back to using a regular BigQuery query to fetch data.
2222
RUN uv pip uninstall --system google-cloud-bigquery-storage
2323

24-
# NOTE(herbison): uv fails to install this for some reason
25-
RUN pip install git+https://github.com/Kaggle/learntools
24+
# uv cannot install this in requirements.txt without --no-build-isolation
25+
# to avoid affecting the larger build, we'll post-install it.
26+
RUN uv pip install --no-build-isolation --system "git+https://github.com/Kaggle/learntools"
2627

27-
# We install an incompatible pair of libs (shapely<, libpysal==4.9.2) so we can't put this one in the requirements.txt
28+
# b/328788268 We install an incompatible pair of libs (shapely<2, libpysal==4.9.2) so we can't put this one in the requirements.txt
2829
RUN uv pip install --system "libpysal==4.9.2"
2930

3031
# Adding non-package dependencies:

kaggle_requirements.txt

Lines changed: 15 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,5 @@
1-
altair>=5.4.0
1+
# Please keep this in alphabetical order
2+
Altair>=5.4.0
23
Babel
34
Boruta
45
Cartopy
@@ -10,7 +11,8 @@ Pympler
1011
Rtree
1112
shapely<2
1213
SimpleITK
13-
TPOT
14+
# b/302136621: Fix eli5 import for learntools, newer version require scikit-learn > 1.3
15+
TPOT==0.12.1
1416
Theano
1517
Wand
1618
annoy
@@ -80,6 +82,8 @@ mamba
8082
mlcrate
8183
mne
8284
mpld3
85+
# b/274619697: learntools requires a specific nbconvert right now
86+
nbconvert==6.4.5
8387
nbdev
8488
nilearn
8589
olefile
@@ -115,13 +119,19 @@ qtconsole
115119
ray
116120
rgf-python
117121
s3fs
118-
scikit-learn-intelex
122+
# b/302136621: Fix eli5 import for learntools
123+
scikit-learn==1.2.2
124+
# Scikit-learn accelerated library for x86
125+
scikit-learn-intelex>=2023.0.1
119126
scikit-multilearn
120127
scikit-optimize
121128
scikit-plot
122129
scikit-surprise
130+
# Also pinning seaborn for learntools
131+
seaborn==0.12.2
123132
git+https://github.com/facebookresearch/segment-anything.git
124-
shap
133+
# b/329869023 shap 0.45.0 breaks learntools
134+
shap==0.44.1
125135
squarify
126136
tensorflow-cloud
127137
tensorflow-io
@@ -134,6 +144,7 @@ tsfresh
134144
vtk
135145
wandb
136146
wavio
147+
# b/350573866: xgboost v2.1.0 breaks learntools
137148
xgboost==2.0.3
138149
xvfbwrapper
139150
ydata-profiling

0 commit comments

Comments
 (0)