py312

Author: calderjo

Dockerfile.tmpl

@@ -41,7 +41,9 @@ RUN uv pip install \
 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
+# RUN uv pip install --system --force-reinstall --index-url https://pypi.anaconda.org/intel/simple numpy==2.0
+# RUN uv pip install --system --force-reinstall -i https://pypi.anaconda.org/intel/simple numpy
+# https://software.repos.intel.com/python/pypi
 
 # newer daal4py requires tbb>=2022, but libpysal is downgrading it for some reason
 RUN uv pip install --system "tbb>=2022" "libpysal==4.9.2"
@@ -57,8 +59,7 @@ ADD clean-layer.sh  /tmp/clean-layer.sh
 ADD patches/nbconvert-extensions.tpl /opt/kaggle/nbconvert-extensions.tpl
 ADD patches/template_conf.json /opt/kaggle/conf.json
 
-# /opt/conda/lib/python3.11/site-packages
-ARG PACKAGE_PATH=/usr/local/lib/python3.11/dist-packages
+ARG PACKAGE_PATH=/usr/local/lib/python3.12/dist-packages
 
 # Install GPU-specific non-pip packages.
 {{ if eq .Accelerator "gpu" }}
@@ -168,7 +169,7 @@ ADD patches/kaggle_gcp.py \
 
 # Figure out why this is in a different place?
 # Found by doing a export PYTHONVERBOSE=1 and then running python and checking for where it looked for it.
-ADD patches/sitecustomize.py /usr/lib/python3.11/sitecustomize.py
+ADD patches/sitecustomize.py /usr/lib/python3.12/sitecustomize.py
 
 ARG GIT_COMMIT=unknown \
     BUILD_DATE=unknown

config.txt

@@ -1,4 +1,4 @@
 BASE_IMAGE=us-docker.pkg.dev/colab-images/public/runtime
-BASE_IMAGE_TAG=release-colab_20250725-060057_RC00
+BASE_IMAGE_TAG=release-colab-external_20251024-060052_RC00
 CUDA_MAJOR_VERSION=12
 CUDA_MINOR_VERSION=5

kaggle_requirements.txt

@@ -7,11 +7,10 @@ PyArabic
 PyUpSet
 Pympler
 Rtree
-shapely<2
+shapely
 SimpleITK
 # b/302136621: Fix eli5 import for learntools, newer version require scikit-learn > 1.3
 TPOT==0.12.1
-Theano
 Wand
 annoy
 arrow
@@ -29,8 +28,6 @@ deap
 dipy
 docker
 easyocr
-# b/302136621: Fix eli5 import for learntools
-eli5
 emoji
 fastcore
 # b/445960030: Requires a newer version of fastai than the currently used base image.
@@ -81,7 +78,7 @@ libpysal<=4.9.2
 lime
 line_profiler
 mamba
-matplotlib<3.8
+matplotlib
 mlcrate
 mne
 mpld3
@@ -107,11 +104,9 @@ preprocessing
 pudb
 pyLDAvis
 pycryptodome
-pydegensac
 pydicom
 pyemd
 pyexcel-ods
-pymc3
 pymongo
 pypdf
 pytesseract
@@ -123,19 +118,15 @@ qtconsole
 ray
 rgf-python
 s3fs
-# b/302136621: Fix eli5 import for learntools
-scikit-learn==1.2.2
+scikit-learn
 # Scikit-learn accelerated library for x86
 scikit-learn-intelex>=2023.0.1
 scikit-multilearn
 scikit-optimize
 scikit-plot
 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
-shap==0.44.1
+seaborn
+shap
 squarify
 tensorflow-cloud
 tensorflow-io