diff --git a/lib/matplotlib/backends/backend_nbagg.py b/lib/matplotlib/backends/backend_nbagg.py index 61ba9d58fd75..146dd91dc8cc 100644 --- a/lib/matplotlib/backends/backend_nbagg.py +++ b/lib/matplotlib/backends/backend_nbagg.py @@ -14,7 +14,12 @@ import tornado.ioloop from IPython.display import display, Javascript, HTML -from IPython.kernel.comm import Comm +try: + # Jupyter/IPython 4.x or later + from ipykernel.comm import Comm +except ImportError: + # Jupyter/IPython 3.x or earlier + from IPython.kernel.comm import Comm from matplotlib import rcParams from matplotlib.figure import Figure