Mock frameworks in pydocs

apache · Nov 29, 2023 · 51ff18b · 51ff18b
1 parent ab60eb3
commit 51ff18b
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/sdks/python/scripts/generate_pydoc.sh b/sdks/python/scripts/generate_pydoc.sh
@@ -133,7 +133,9 @@ autodoc_inherit_docstrings = False
 autodoc_member_order = 'bysource'
 autodoc_mock_imports = ["tensorrt", "cuda", "torch",
     "onnxruntime", "onnx", "tensorflow", "tensorflow_hub",
-    "tensorflow_transform", "tensorflow_metadata", "transformers"]
+    "tensorflow_transform", "tensorflow_metadata", "transformers", "tensorflow_text",
+    "sentence_transformers",
+    ]
 
 # Allow a special section for documenting DataFrame API
 napoleon_custom_sections = ['Differences from pandas']