support #@hide, #@hide_output, #@hide_code to hide notebook code and …

…output
d2l-ai · Oct 23, 2020 · 89b8934 · 89b8934
1 parent 52f80cc
commit 89b8934
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/d2lbook/rst.py b/d2lbook/rst.py
@@ -41,9 +41,10 @@ def _process_nb(nb):
     # hide/show
     for cell in new_cells:
         if cell.cell_type == 'code':
-            if '# hide outputs' in cell.source.lower():
+            src = cell.source.lower()
+            if '# hide outputs' in src or '#@hide' in src or '#@hide_output' in src:
                 cell.outputs = []
-            if '# hide code' in cell.source.lower():
+            if '# hide code' in src or '#@hide' in src or '#@hide_code' in src:
                 cell.source = ''
     return notebook.create_new_notebook(nb, new_cells)