Try adding path modification as first notebook cell

Author: samanklesaria

docs/source/conf.py

@@ -22,8 +22,10 @@
 import sys
 import warnings
 from datetime import datetime
+from pathlib import Path
 
 sys.path.insert(0, os.path.abspath("."))
+TEST_DIR = str(Path('..', '..', 'test').resolve())
 
 
 import pytorch_sphinx_theme
@@ -157,9 +159,12 @@ def reset_mpl(gallery_conf, fname):
     **_get_pattern(),
     "backreferences_dir": "gen_modules/backreferences",
     "promote_jupyter_magic": True,
-    "first_notebook_cell": None,
     "doc_module": ("torchaudio",),
     "reset_modules": (reset_mpl, "seaborn"),
+    'first_notebook_cell': (
+        "import sys\n"
+        f"sys.path.insert(0, {TEST_DIR})"
+    )
 }
 autosummary_generate = True