From 51ff18ba4db584035254e0fa40df42b47d3bf157 Mon Sep 17 00:00:00 2001 From: Anand Inguva Date: Wed, 29 Nov 2023 10:51:17 -0500 Subject: [PATCH] Mock frameworks in pydocs --- sdks/python/scripts/generate_pydoc.sh | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/sdks/python/scripts/generate_pydoc.sh b/sdks/python/scripts/generate_pydoc.sh index 06ad06320fcf..8d5b43167dd1 100755 --- 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']