mockimport jaxtyping

liopeer · Sep 2, 2023 · 359c4aa · 359c4aa
1 parent 8771316
commit 359c4aa
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/docs/source/conf.py b/docs/source/conf.py
@@ -45,7 +45,7 @@
 autodoc_inherit_docstring = True
 templates_path = ["_templates"]
 
-autodoc_mock_imports = ["torch"]
+autodoc_mock_imports = ["torch, jaxtyping"]
 
 source_suffix = {
     ".rst": "restructuredtext",