Fix docs build

Author: benjeffery

docs/ancestry.md

@@ -16,7 +16,8 @@ kernelspec:
 :tags: [remove-cell]
 
 import msprime
-from IPython.display import SVG, set_matplotlib_formats
+from IPython.display import SVG
+from matplotlib_inline.backend_inline import set_matplotlib_formats
 from matplotlib import pyplot as plt
 import numpy as np
 import networkx as nx

docs/demography.md

@@ -18,7 +18,8 @@ kernelspec:
 import msprime
 import numpy as np
 import matplotlib.pyplot as plt
-from IPython.display import SVG, set_matplotlib_formats
+from IPython.display import SVG
+from matplotlib_inline.backend_inline import set_matplotlib_formats
 
 set_matplotlib_formats("svg")
 plt.rcParams["figure.figsize"] = (10,7)

docs/pedigrees.md

@@ -18,7 +18,8 @@ kernelspec:
 import io
 import msprime
 import tskit
-from IPython.display import SVG, set_matplotlib_formats
+from IPython.display import SVG
+from matplotlib_inline.backend_inline import set_matplotlib_formats
 from matplotlib import pyplot as plt
 import numpy as np
 import networkx as nx

docs/rate_maps.md

@@ -17,7 +17,8 @@ kernelspec:
 import msprime
 import numpy as np
 import io
-from IPython.display import SVG, set_matplotlib_formats
+from IPython.display import SVG
+from matplotlib_inline.backend_inline import set_matplotlib_formats
 from matplotlib import pyplot as plt
 
 set_matplotlib_formats("svg")