From 2dd9f434b0b594be962cbb77ef2562f75f471155 Mon Sep 17 00:00:00 2001
From: Jonathan Calderon Chavez <chavezcalderon@google.com>
Date: Thu, 14 Mar 2024 20:33:40 +0000
Subject: [PATCH] pin tfdf new ver. not comp

---
 Dockerfile.tmpl | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/Dockerfile.tmpl b/Dockerfile.tmpl
index 5b078767..16423203 100644
--- a/Dockerfile.tmpl
+++ b/Dockerfile.tmpl
@@ -198,7 +198,8 @@ RUN JAXVER=$(pip freeze | grep -e "^jax==") && \
         "${JAXVER}" && \
     /tmp/clean-layer.sh
 
-RUN apt-get install -y default-jre && \
+RUN apt-get update && \
+    apt-get install -y default-jre && \
     /tmp/clean-layer.sh
 
 RUN pip install -f http://h2o-release.s3.amazonaws.com/h2o/latest_stable_Py.html h2o && /tmp/clean-layer.sh
@@ -214,7 +215,7 @@ RUN pip install \
     /tmp/clean-layer.sh
 
 # b/318672158 Use simply tensorflow_decision_forests on next release, expected with tf 2.16
-RUN pip install tensorflow_decision_forests --no-deps && \
+RUN pip install tensorflow_decision_forests==1.8.1 --no-deps && \
     /tmp/clean-layer.sh
 
 RUN chmod +x /tmp/keras_patch.sh && \