Lingwave · bryce13950 · Oct 22, 2023 · Oct 22, 2023 · Oct 22, 2023 · Oct 22, 2023
diff --git a/demos/Main_Demo.ipynb b/demos/Main_Demo.ipynb
diff --git a/demos/doc_sanitize.cfg b/demos/doc_sanitize.cfg
@@ -0,0 +1,11 @@
+[regex1]
+regex: \d{1,2}/\d{1,2}/\d{2,4}
+replace: DATE-STAMP
+
+[regex2]
+regex: \d{2}:\d{2}:\d{2}
+replace: TIME-STAMP
+
+[regex3]
+regex: 0[xX][0-9a-fA-F]+
+replace: HEX-CODE
diff --git a/docs/make_docs.py b/docs/make_docs.py
@@ -116,7 +116,10 @@ def generate_model_table(_app: Optional[Any] = None):
 def copy_demos(_app: Optional[Any] = None):
     """Copy demo notebooks to the generated directory."""
     copy_to_dir = GENERATED_DIR / "demos"
-    notebooks_to_copy = ["Exploratory_Analysis_Demo.ipynb"]
+    notebooks_to_copy = [
+        "Exploratory_Analysis_Demo.ipynb",
+        "Main_Demo.ipynb",
+    ]
 
     if copy_to_dir.exists():
         shutil.rmtree(copy_to_dir)

diff --git a/docs/source/index.md b/docs/source/index.md
@@ -42,6 +42,7 @@ generated/model_properties_table.md
 content/tutorials
 content/citation
 content/contributing
+generated/demos/Main_Demo
 generated/demos/Exploratory_Analysis_Demo
 ```
 

diff --git a/makefile b/makefile
@@ -18,7 +18,8 @@ docstring-test:
 	poetry run pytest transformer_lens/
 
 notebook-test:
-	poetry run pytest demos/Exploratory_Analysis_Demo.ipynb
+	poetry run pytest --nbval-sanitize-with demos/doc_sanitize.cfg demos/Main_Demo.ipynb 
+	poetry run pytest --nbval-sanitize-with demos/doc_sanitize.cfg demos/Exploratory_Analysis_Demo.ipynb
 
 test:
 	make unit-test

diff --git a/poetry.lock b/poetry.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -66,6 +66,7 @@ nbval = "^0.10.0"
 
 [tool.poetry.group.jupyter.dependencies]
 jupyterlab = ">=3.5.0"
+ipywidgets = "^8.1.1"
 
 [tool.poetry.group.docs.dependencies]
 sphinx = {version = "5.2.3" }