[Docs] Fix Unmocked Imports (#3275)

This commit is contained in:
Roger Wang 2024-03-08 09:58:01 -08:00 committed by GitHub
parent 1ece1ae829
commit 99c3cfb83c
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -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: