-
Notifications
You must be signed in to change notification settings - Fork 27.6k
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Add type hints for pytorch models (final batch) #25750
Add type hints for pytorch models (final batch) #25750
Conversation
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
LGTM!
Thank you again for the diligent work on the type hints, by the way - your PRs were really clean and required very little work to review, and they really helped push that project! |
The docs for this PR live here. All of your documentation changes will be reflected on that endpoint. |
Co-authored-by: Matt <Rocketknight1@users.noreply.github.com>
Thank you very much, really appreciated!! TBH it has been a great learning experience for me, so thanks for the opportunity and the patience with the explanations |
My apologies, in advance, for stepping in here. |
Gema, thanks for the kind words 🤗 |
* Add type hints for table_transformer * Add type hints to Timesformer model * Add type hints to Timm Backbone model * Add type hints to TVLT family models * Add type hints to Vivit family models * Use the typing instance instead of the python builtin. * Fix the `replace_return_docstrings` decorator for Vivit model Co-authored-by: Matt <Rocketknight1@users.noreply.github.com> --------- Co-authored-by: Matt <Rocketknight1@users.noreply.github.com>
* Add type hints for table_transformer * Add type hints to Timesformer model * Add type hints to Timm Backbone model * Add type hints to TVLT family models * Add type hints to Vivit family models * Use the typing instance instead of the python builtin. * Fix the `replace_return_docstrings` decorator for Vivit model Co-authored-by: Matt <Rocketknight1@users.noreply.github.com> --------- Co-authored-by: Matt <Rocketknight1@users.noreply.github.com>
* Add type hints for table_transformer * Add type hints to Timesformer model * Add type hints to Timm Backbone model * Add type hints to TVLT family models * Add type hints to Vivit family models * Use the typing instance instead of the python builtin. * Fix the `replace_return_docstrings` decorator for Vivit model Co-authored-by: Matt <Rocketknight1@users.noreply.github.com> --------- Co-authored-by: Matt <Rocketknight1@users.noreply.github.com>
What does this PR do?
Addresses the last type hints for pytorch models in #16059:
TableTransformerForObjectDetection
TableTransformerModel
TimesformerModel
TimmBackbone
TvltForAudioVisualClassification
TvltForPreTraining
TvltModel
VivitForVideoClassification
VivitModel
Who can review?
@Rocketknight1 please