From 06e74c199792035277465b7e3b01b5a61a3786d0 Mon Sep 17 00:00:00 2001 From: Johnny Date: Mon, 22 Jun 2026 18:27:34 +0000 Subject: [PATCH] new base image --- Dockerfile.tmpl | 13 ++----------- tests/test_tensorflow_decision_forests.py | 18 ------------------ 2 files changed, 2 insertions(+), 29 deletions(-) delete mode 100644 tests/test_tensorflow_decision_forests.py diff --git a/Dockerfile.tmpl b/Dockerfile.tmpl index 86c5426f..d9d38935 100644 --- a/Dockerfile.tmpl +++ b/Dockerfile.tmpl @@ -1,7 +1,7 @@ {{ if eq .Accelerator "gpu" }} -FROM us-docker.pkg.dev/colab-images/public/runtime:release-colab-external-images_20260416-060047_RC00 +FROM us-docker.pkg.dev/colab-images/public/runtime:release-colab-external-images_20260514-060047_RC00 {{ else }} -FROM us-docker.pkg.dev/colab-images/public/cpu-runtime:release-colab-external-images_20260416-060047_RC00 +FROM us-docker.pkg.dev/colab-images/public/cpu-runtime:release-colab-external-images_20260514-060047_RC00 {{ end}} ADD kaggle_requirements.txt /kaggle_requirements.txt @@ -28,8 +28,6 @@ RUN uv pip install --no-build-isolation --no-cache --system "git+https://github. # b/468367647: Unpin protobuf, version greater than v5.29.5 causes issues with numerous packages RUN uv pip install --system --force-reinstall --no-cache --no-deps torchtune RUN uv pip install --system --force-reinstall --no-cache "protobuf==5.29.5" -# b/493600019: Colab base image ships numba that does not support NumPy 2.4; upgrade to latest. -RUN uv pip install --system --force-reinstall --no-cache numba # Adding non-package dependencies: ADD clean-layer.sh /tmp/clean-layer.sh @@ -40,14 +38,7 @@ ARG PACKAGE_PATH=/usr/local/lib/python3.12/dist-packages # Install GPU-specific non-pip packages. {{ if eq .Accelerator "gpu" }} -# b/493600019: numba-cuda v0.30.0 fixes np.trapz removal in NumPy 2.4 but requires libcudart.so (GPU only). -RUN uv pip install --system --force-reinstall --no-cache numba-cuda RUN uv pip install --system --no-cache "pycuda" -{{ else }} -# b/493600019: On CPU, remove numba-cuda shipped by the Colab base image. Newer numba-cuda -# depends on cuda-bindings which crashes at import without libcudart.so. Packages like -# tsfresh/stumpy that import numba.cuda will fall back gracefully without it. -RUN uv pip uninstall --system numba-cuda 2>/dev/null || true {{ end }} diff --git a/tests/test_tensorflow_decision_forests.py b/tests/test_tensorflow_decision_forests.py deleted file mode 100644 index aeeb2e7d..00000000 --- a/tests/test_tensorflow_decision_forests.py +++ /dev/null @@ -1,18 +0,0 @@ -import unittest - -import numpy as np -import pandas as pd -import tensorflow_decision_forests as tfdf - -class TestTensorflowDecisionForest(unittest.TestCase): - def test_fit(self): - train_df = pd.read_csv("/input/tests/data/train.csv") - - # Convert the dataset into a TensorFlow dataset. - train_ds = tfdf.keras.pd_dataframe_to_tf_dataset(train_df, label="label") - - # Train the model - model = tfdf.keras.RandomForestModel(num_trees=1) - model.fit(train_ds) - - self.assertEqual(1, model.count_params()) \ No newline at end of file