limit ref substitution to md cells

TUW-GEO · Aug 14, 2024 · f384f9a · f384f9a
1 parent 4936822
commit f384f9a
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/src/eo_datascience/clean_nb.py b/src/eo_datascience/clean_nb.py
@@ -40,7 +40,8 @@ def convert_refs(dir="./notebooks", save=True):
         nb = nbformat.read(nb_path, as_version=4)
         for i in range(len(nb.cells)):
             if i != 0:
-                nb.cells[i].source = nb.cells[i].source.replace(r"[@", r"{cite}`").replace(r"]", r"`")
+                if nb.cells[i]["cell_type"] == "markdown":
+                    nb.cells[i].source = nb.cells[i].source.replace(r"[@", r"{cite}`").replace(r"]", r"`")
 
         # Save the notebook
         if save:

diff --git a/src/eo_datascience/render_sfinx_toc.py b/src/eo_datascience/render_sfinx_toc.py
@@ -16,4 +16,4 @@
 )
 
 with open('_toc.yml', 'w+') as ff:
-    yaml.dump(toc, ff)
+    yaml.dump(toc, ff)