Mock packages instead of installing for pydocs (#29692)

apache · Dec 9, 2023 · aafa3b8 · aafa3b8
1 parent 4891a81
commit aafa3b8
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 5 deletions.
diff --git a/sdks/python/scripts/generate_pydoc.sh b/sdks/python/scripts/generate_pydoc.sh
@@ -133,7 +133,7 @@ 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", "xgboost", "datatable", "transformers"]
 
 # Allow a special section for documenting DataFrame API
 napoleon_custom_sections = ['Differences from pandas']

diff --git a/sdks/python/tox.ini b/sdks/python/tox.ini
@@ -148,10 +148,6 @@ deps =
   sphinx_rtd_theme==0.4.3
   docutils<0.18
   Jinja2==3.0.3 # TODO(https://github.com/apache/beam/issues/21587): Sphinx version is too old.
-  torch
-  xgboost<=1.7.6
-  datatable==1.0.0
-  transformers
 commands =
   time {toxinidir}/scripts/generate_pydoc.sh