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

http://b/308525631

Author: Philmod

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 \