From 1318e3374920defb10450a3ef27dd10a4a372b40 Mon Sep 17 00:00:00 2001
From: Nesh Devanathan <neshdev@google.com>
Date: Fri, 18 Aug 2023 04:01:20 +0000
Subject: [PATCH] h2o package needs java jre

---
 Dockerfile.tmpl | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/Dockerfile.tmpl b/Dockerfile.tmpl
index a5cb8ef1..4d75a47a 100644
--- a/Dockerfile.tmpl
+++ b/Dockerfile.tmpl
@@ -183,6 +183,8 @@ RUN JAXVER=$(pip freeze | grep -e "^jax==") && \
         flax \
         "${JAXVER}"
 
+RUN apt-get install -y default-jre
+
 RUN pip install -f http://h2o-release.s3.amazonaws.com/h2o/latest_stable_Py.html h2o && /tmp/clean-layer.sh
 
 RUN pip install \
@@ -251,7 +253,6 @@ RUN pip install scipy \
         datashader \
         # Boruta (python implementation)
         Boruta && \
-
     apt-get install -y graphviz && pip install graphviz && \
     # Pandoc is a dependency of deap
     apt-get install -y pandoc && \
@@ -473,6 +474,7 @@ RUN pip install bleach \
         pyarrow \
         feather-format \
         fastai
+
 RUN python -m spacy download en_core_web_sm && python -m spacy download en_core_web_lg && \
     apt-get update && apt-get install -y ffmpeg && \
     /tmp/clean-layer.sh