Install latest version of seaborn which fixes heatmap (#1315)

http://b/308525631
Kaggle · Oct 31, 2023 · 71c35fe · 71c35fe
1 parent 4d71248
commit 71c35fe
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/Dockerfile.tmpl b/Dockerfile.tmpl
@@ -178,8 +178,12 @@ RUN export PATH=/usr/local/cuda/bin:$PATH && \
     /tmp/clean-layer.sh
 {{ end }}
 
+# Make sure to install at least the version 0.13.0 of seaborn
+# which has a fix for heatmap (b/308525631)
 RUN JAXVER=$(pip freeze | grep -e "^jax==") && \
-    pip install seaborn python-dateutil dask igraph \
+    pip install \
+        seaborn>=0.13.0 \
+        python-dateutil dask igraph \
         pyyaml joblib husl geopy mne pyshp \
         pandas \
         polars \