added import

docs/scripts/patch_devsite_toc.py

@@ -61,6 +61,7 @@ def __init__(self, dir_name, index_file_name):
         self.dir_name = dir_name
         self.index_file_name = index_file_name
         index_file_path = os.path.join(dir_name, index_file_name)
+        # find set of files referenced by the index file
         with open(index_file_path, "r") as f:
             self.title = f.readline().replace("\n", "")
             in_toc = False

noxfile.py

@@ -428,6 +428,7 @@ def docfx(session):
     # Customization: Add extra sections to the table of contents for the Standard vs Async clients
     import sys
 
+    session.install("pyyaml")
     # import local module at docs/scripts/patch_devsite_toc.py
     sys.path.append(os.path.join("docs", "scripts"))
     import patch_devsite_toc

owlbot.py

@@ -217,8 +217,11 @@ def mypy(session):
 docfx_postprocess = """
     # Customization: Add extra sections to the table of contents for the Standard vs Async clients
     import sys
-    sys.path.append(os.path.abspath(os.path.join("docs", "scripts")))
-    from docs.scripts import patch_devsite_toc
+
+    session.install("pyyaml")
+    # import local module at docs/scripts/patch_devsite_toc.py
+    sys.path.append(os.path.join("docs", "scripts"))
+    import patch_devsite_toc
     toc_path = os.path.join("_build", "html", "docfx_yaml", "toc.yml")
     custom_sections = [
         patch_devsite_toc.TocSection(