testeeest

Author: calderjo

Dockerfile.tmpl

@@ -32,12 +32,6 @@ RUN uv pip install --system --force-reinstall --prerelease=allow kagglehub[panda
 # to avoid affecting the larger build, we'll post-install it.
 RUN uv pip install --no-build-isolation --system "git+https://github.com/Kaggle/learntools"
 
-# b/408281617: Torch is adamant that it can not install cudnn 9.3.x, only 9.1.x, but Tensorflow can only support 9.3.x.
-# This conflict causes a number of package downgrades, which are handled in this command
-RUN uv pip install --system --force-reinstall --extra-index-url https://pypi.nvidia.com "cuml-cu12==25.2.1" \
-    "nvidia-cudnn-cu12==9.3.0.75"
-RUN uv pip install --system --force-reinstall "pynvjitlink-cu12==0.5.2"
-
 # b/385145217 Latest Colab lacks mkl numpy, install it.
 RUN uv pip install --system --force-reinstall -i https://pypi.anaconda.org/intel/simple numpy
 

kaggle_requirements.txt

@@ -121,6 +121,7 @@ qtconsole
 ray
 rgf-python
 s3fs
+# b/302136621: Fix eli5 import for learntools
 scikit-learn==1.2.2
 # Scikit-learn accelerated library for x86
 scikit-learn-intelex>=2023.0.1
@@ -131,7 +132,7 @@ scikit-surprise
 # Also pinning seaborn for learntools
 seaborn==0.12.2
 git+https://github.com/facebookresearch/segment-anything.git
-# b/329869023 shap 0.45.0 breaks learntools
+# b/329869023: shap 0.45.0 breaks learntools
 shap==0.44.1
 squarify
 tensorflow-cloud