Add highlight_extra_classes to outputs

danielfrg · Mar 19, 2023 · 401ca0d · 401ca0d
1 parent b2c4ce1
commit 401ca0d
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/mkdocs_jupyter/nbconvert2.py b/mkdocs_jupyter/nbconvert2.py
@@ -155,6 +155,12 @@ def custom_clean_html(element):
             },
         )
 
+    if highlight_extra_classes:
+        content = content.replace(
+            "jp-OutputArea-output",
+            f"jp-OutputArea-output {highlight_extra_classes}",
+        )
+        return content
     return content
 
 

diff --git a/mkdocs_jupyter/tests/mkdocs/material-with-nbs.yml b/mkdocs_jupyter/tests/mkdocs/material-with-nbs.yml
@@ -10,8 +10,7 @@ nav:
 plugins:
   - mkdocs-jupyter:
       include_source: true
-      highlight_extra_classes: "custom-css-classes"
-      include_requirejs: true
+      highlight_extra_classes: not-prose
 
 markdown_extensions:
   - toc: