diff --git a/docs/book/post_build.py b/docs/book/post_build.py index 4185033..1abc3a2 100644 --- a/docs/book/post_build.py +++ b/docs/book/post_build.py @@ -2,6 +2,7 @@ import os import shutil import subprocess import sys +import re def main(): @@ -43,7 +44,7 @@ def build_and_copy_demo(category, md_name): html = f.read() head_split = html.split("") target_head = head_split[1].split("")[0] - body_split = html.split("")[1].split("") + body_split = re.split("]*>", html)[1].split("") target_body = body_split[0] with open(book_html_path, "w") as f: