fix tfdf

Author: djherbis

Dockerfile.tmpl

@@ -32,6 +32,10 @@ RUN uv pip uninstall --system google-cloud-bigquery-storage
 # 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/385161357 Latest Colab uses tf 2.17.1, but tf decision forests only has a version for 2.17.0.
+# Instead, we'll install tfdf with its deps and hope that 2.17.0 compat tfdf works with tf 2.17.1.
+RUN uv pip install --system --no-deps tensorflow-decision-forests==1.10.0 wurlitzer==3.1.1 ydf==0.9.0
+
 # b/385145217 Latest Colab lacks mkl numpy, install it.
 RUN uv pip install --system --force-reinstall -i https://pypi.anaconda.org/intel/simple numpy
 

tests/test_tensorflow_decision_forests.py

@@ -4,7 +4,6 @@
 import pandas as pd
 import tensorflow_decision_forests as tfdf
 
-@unittest.skip("b/385161357: tensorflow_decision_forests is incompatible with tensorflow 2.17.1")
 class TestTensorflowDecisionForest(unittest.TestCase):
     def test_fit(self):
         train_df = pd.read_csv("/input/tests/data/train.csv")