Pin matplolib as upgrade of seaborn has conflict (#1316)

http://b/308525631
Kaggle · Nov 1, 2023 · 4f923e3 · 4f923e3
1 parent 71c35fe
commit 4f923e3
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 5 deletions.
diff --git a/Dockerfile.tmpl b/Dockerfile.tmpl
@@ -178,11 +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)
+# (b/308525631) Pin Matplotlib until seaborn can be upgraded
+# to >0.13.0 (now it's stuck by a package conflict with ydata-profiling 4.5.1).
 RUN JAXVER=$(pip freeze | grep -e "^jax==") && \
-    pip install \
-        seaborn>=0.13.0 \
+    pip install --upgrade \
+        "matplotlib<3.8.0" \
+        seaborn \
         python-dateutil dask igraph \
         pyyaml joblib husl geopy mne pyshp \
         pandas \

diff --git a/tests/test_seaborn.py b/tests/test_seaborn.py
@@ -7,7 +7,7 @@
 class TestSeaborn(unittest.TestCase):
     # Fails if seaborn gets downgraded by other package installations.
     def test_version(self):
-        self.assertGreaterEqual(StrictVersion(sns.__version__), StrictVersion("0.9.0"))
+        self.assertGreaterEqual(StrictVersion(sns.__version__), StrictVersion("0.12.0"))
 
 
     def test_option(self):