Fix rapids_related_examples extension

rapidsai · Aug 2, 2023 · 2296c6a · 2296c6a
1 parent df105af
commit 2296c6a
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/extensions/rapids_related_examples.py b/extensions/rapids_related_examples.py
@@ -72,6 +72,8 @@ def get_title_for_notebook(path: str) -> str:
         if cell["cell_type"] == "markdown":
             cell_source = MarkdownIt().parse(cell["source"])
             for i, token in enumerate(cell_source):
+                if i == len(cell_source) - 1:  # no next_token
+                    continue
                 next_token = cell_source[i + 1]
                 if (
                     token.type == "heading_open"