diff --git a/pymc3/tests/test_examples.py b/pymc3/tests/test_examples.py index 7d431dfd07..9afdac1079 100644 --- a/pymc3/tests/test_examples.py +++ b/pymc3/tests/test_examples.py @@ -20,10 +20,14 @@ import pytest import theano from pymc3.theanof import floatX +from packaging import version from .helpers import SeededTest -matplotlib.use('Agg', warn=False) +if version.parse(matplotlib.__version__) < version.parse('3.3'): + matplotlib.use('Agg', warn=False) +else: + matplotlib.use('Agg') def get_city_data():