Fix for upcoming nbconvert

Code from spatialaudio/nbsphinx#344 with thanks.
matthew-brett · Jan 2, 2020 · e91fed3 · e91fed3
1 parent e4d0e3e
commit e91fed3
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/nb2plots/from_notebook.py b/nb2plots/from_notebook.py
@@ -13,8 +13,8 @@
 
 
 # Template to label code and output and plot blocks
-dl = DictLoader({'rst_plots.tpl': """\
-{%- extends 'rst.tpl' -%}
+dl = DictLoader({'rst_plots.tpl': r"""\
+{%- extends '__RST_DEFAULT_TEMPLATE__' -%}
 
 {% block input %}
 {%- if cell.source.strip() | has_mpl_inline -%}
@@ -50,7 +50,8 @@
 {{ output.data['text/plain'] | ellipse_mpl | indent }}
 ##END_OUT_END##
 {%- endblock data_text -%}
-"""})
+""".replace('__RST_DEFAULT_TEMPLATE__',
+            nbconvert.RSTExporter().template_file)})
 
 
 def has_mpl_inline(code):