Skip to content

Commit

Permalink
unpin transformers for pydoc (#28436)
Browse files Browse the repository at this point in the history
  • Loading branch information
riteshghorse authored Sep 13, 2023
1 parent 1032533 commit 5cb55b2
Show file tree
Hide file tree
Showing 2 changed files with 4 additions and 2 deletions.
4 changes: 3 additions & 1 deletion sdks/python/scripts/generate_pydoc.sh
Original file line number Diff line number Diff line change
Expand Up @@ -131,7 +131,9 @@ release = version
autoclass_content = 'both'
autodoc_inherit_docstrings = False
autodoc_member_order = 'bysource'
autodoc_mock_imports = ["tensorrt", "cuda", "torch", "onnxruntime", "onnx", "tensorflow", "tensorflow_hub", "tensorflow_transform", "tensorflow_metadata"]
autodoc_mock_imports = ["tensorrt", "cuda", "torch",
"onnxruntime", "onnx", "tensorflow", "tensorflow_hub",
"tensorflow_transform", "tensorflow_metadata", "transformers"]
# Allow a special section for documenting DataFrame API
napoleon_custom_sections = ['Differences from pandas']
Expand Down
2 changes: 1 addition & 1 deletion sdks/python/tox.ini
Original file line number Diff line number Diff line change
Expand Up @@ -165,7 +165,7 @@ deps =
torch
xgboost<=1.7.6
datatable==1.0.0
transformers<=4.31.0
transformers
commands =
time {toxinidir}/scripts/generate_pydoc.sh

Expand Down

0 comments on commit 5cb55b2

Please sign in to comment.