diff --git a/Dockerfile.tmpl b/Dockerfile.tmpl index 1b63837d..7561a700 100644 --- a/Dockerfile.tmpl +++ b/Dockerfile.tmpl @@ -25,8 +25,8 @@ RUN uv pip uninstall --system google-cloud-bigquery-storage # b/394382016: sigstore (dependency of kagglehub) requires a prerelease packages, installing separate. # b/408284143: google-cloud-automl 2.0.0 introduced incompatible API changes, need to pin to 1.0.1, # installed outside of kaggle_requirements.txt due to requiring an incompatibile version of protobuf. -RUN uv pip install --system --force-reinstall --prerelease=allow kagglehub[pandas-datasets,hf-datasets,signing]>=0.3.12 \ - google-cloud-automl==1.0.1 +RUN uv pip install --system --force-reinstall --prerelease=allow "kagglehub[pandas-datasets,hf-datasets,signing]>=0.3.12" \ + "google-cloud-automl==1.0.1" # uv cannot install this in requirements.txt without --no-build-isolation # to avoid affecting the larger build, we'll post-install it. diff --git a/config.txt b/config.txt index 61395f5e..af541652 100644 --- a/config.txt +++ b/config.txt @@ -1,4 +1,4 @@ BASE_IMAGE=us-docker.pkg.dev/colab-images/public/runtime -BASE_IMAGE_TAG=release-colab_20250626-060053_RC00 +BASE_IMAGE_TAG=release-colab_20250725-060057_RC00 CUDA_MAJOR_VERSION=12 CUDA_MINOR_VERSION=5 diff --git a/kaggle_requirements.txt b/kaggle_requirements.txt index 03c489b4..c711869e 100644 --- a/kaggle_requirements.txt +++ b/kaggle_requirements.txt @@ -1,6 +1,4 @@ # Please keep this in alphabetical order -Altair>=5.4.0 -Babel Boruta Cartopy ImageHash @@ -24,7 +22,6 @@ category-encoders cesium comm cytoolz -dask-expr # Older versions of datasets fail with "Loading a dataset cached in a LocalFileSystem is not supported" # https://stackoverflow.com/questions/77433096/notimplementederror-loading-a-dataset-cached-in-a-localfilesystem-is-not-suppor datasets>=2.14.6 @@ -48,6 +45,8 @@ geojson # geopandas > v0.14.4 breaks learn tools geopandas==v0.14.4 gensim +# b/443054743 +google-adk google-cloud-aiplatform # b/315753846: Unpin translate package. google-cloud-translate==3.12.1 @@ -111,7 +110,6 @@ pyLDAvis pycryptodome pydegensac pydicom -pydub pyemd pyexcel-ods pymc3 @@ -144,18 +142,15 @@ tensorflow-cloud tensorflow-io tensorflow-text tensorflow_decision_forests -timm -torchao torchinfo torchmetrics torchtune transformers>=4.51.0 -triton -tsfresh vtk -wandb wavio # b/350573866: xgboost v2.1.0 breaks learntools xgboost==2.0.3 xvfbwrapper ydata-profiling +# b/443054743: pinned as newer versions requires protobuf > 3.20.3 +ydf==0.9.0