diff --git a/Dockerfile.tmpl b/Dockerfile.tmpl index 4d5344fe..ece2998d 100644 --- 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 index b20cbe27..5d9a76c1 100644 --- 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):