[Doc] Fix Mock Import (#6094)

f1c78138 · Roger Wang · GitHub · 3a86b54f · f1c78138
Unverified Commit f1c78138 authored Jul 03, 2024 by Roger Wang Committed by GitHub Jul 03, 2024
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 0 deletions

docs/source/conf.py docs/source/conf.py +1 -0

No files found.
--- a/docs/source/conf.py
+++ b/docs/source/conf.py
@@ -96,6 +96,7 @@ autodoc_mock_imports = [
    'triton',
    "tqdm",
    "tensorizer",
+    "pynvml",
 ]

 for mock_target in autodoc_mock_imports: