Skip to content

Commit 068add8

Browse files
committed
Add kaggle-benchmarks
1 parent 9aa84ad commit 068add8

2 files changed

Lines changed: 2 additions & 1 deletion

File tree

Dockerfile.tmpl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -27,7 +27,7 @@ RUN uv pip install --no-build-isolation --no-cache --system "git+https://github.
2727
# b/404590350: Ray and torchtune have conflicting cli named `tune`. `ray` is not part of Colab's base image. Re-install `tune` to ensure the torchtune CLI is available by default.
2828
# b/468367647: Unpin protobuf, version greater than v5.29.5 causes issues with numerous packages
2929
RUN uv pip install --system --force-reinstall --no-cache --no-deps torchtune
30-
RUN uv pip install --system --force-reinstall --no-cache "protobuf==5.29.5"
30+
RUN uv pip install --system --force-reinstall --no-cache "protobuf==5.29.6"
3131
# b/493600019: Colab base image ships numba that does not support NumPy 2.4; upgrade to latest.
3232
RUN uv pip install --system --force-reinstall --no-cache numba
3333

kaggle_requirements.txt

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -59,6 +59,7 @@ jupyter_server_proxy
5959
jupyterlab
6060
jupyterlab-lsp
6161
kaggle>=1.8.3
62+
kaggle-benchmarks
6263
kaggle-environments
6364
kagglehub[pandas-datasets,hf-datasets,signing]>=0.4.2
6465
keras-cv

0 commit comments

Comments
 (0)