[Docs] Fix Unmocked Imports (#3275)

2024-03-08 09:58:01 -08:00
parent 1ece1ae829
commit 99c3cfb83c
1 changed files with 9 additions and 2 deletions
--- 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: