[Docs] Fix Unmocked Imports (#3275)

99c3cfb8 · Roger Wang · GitHub · 1ece1ae8 · 99c3cfb8
Unverified Commit 99c3cfb8 authored Mar 08, 2024 by Roger Wang Committed by GitHub Mar 08, 2024
Show whitespace changes
Inline Side-by-side

Showing with 9 additions and 2 deletions

docs/source/conf.py docs/source/conf.py +9 -2

No files found.
--- a/docs/source/conf.py
+++ b/docs/source/conf.py
@@ -72,8 +72,15 @@ html_theme_options = {

 # Mock out external dependencies here.
 autodoc_mock_imports = [
-    "torch", "transformers", "psutil", "prometheus_client", "sentencepiece",
-    "vllm.cuda_utils", "vllm._C"
+    "torch",
+    "transformers",
+    "psutil",
+    "prometheus_client",
+    "sentencepiece",
+    "vllm.cuda_utils",
+    "vllm._C",
+    "numpy",
+    "tqdm",
 ]

 for mock_target in autodoc_mock_imports: