Merge branch 'main' into upgrade-tf2.9

Author: djherbis

Dockerfile.tmpl

@@ -176,6 +176,7 @@ RUN pip install pysal && \
     pip install tensorflow-gcs-config==${TENSORFLOW_VERSION} && \
     # TODO(b/207851560) Upgrade to 0.17.1 once the base image with TensorFlow 2.9.1 is out.
     pip install tensorflow-addons==0.17.0 && \
+    pip install tensorflow_decision_forests==0.2.0 && \
     /tmp/clean-layer.sh
 
 RUN apt-get install -y libfreetype6-dev && \

tests/test_tensorflow_decision_forests.py

@@ -0,0 +1,18 @@
+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())