Fix matplotlib showing the plots instead of saving them in doctests

jupyter · Oct 20, 2023 · 8e11f52 · 8e11f52
1 parent 7b6ec43
commit 8e11f52
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/papyri/gen.py b/papyri/gen.py
@@ -1164,6 +1164,10 @@ def dbg(*args):
         class PapyriDocTestRunner(doctest.DocTestRunner):
             def __init__(slf, *args, **kwargs):
                 super().__init__(*args, **kwargs)
+                import matplotlib
+
+                matplotlib.use("agg")
+
                 slf.figs = []
                 slf.fig_managers = _pylab_helpers.Gcf.get_all_fig_managers()
                 assert (len(slf.fig_managers)) == 0, f"init fail in {qa} {len(slf.fig_managers)}"