[Doc] Fix Mock Import (#6094)

This commit is contained in:
Roger Wang 2024-07-03 00:13:56 -07:00 committed by GitHub
parent 3a86b54fb0
commit f1c78138aa
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -96,6 +96,7 @@ autodoc_mock_imports = [
'triton', 'triton',
"tqdm", "tqdm",
"tensorizer", "tensorizer",
"pynvml",
] ]
for mock_target in autodoc_mock_imports: for mock_target in autodoc_mock_imports: