"vscode:/vscode.git/clone" did not exist on "1fb3c23b41c80d4c6061551ba5ef5dc54637bdba"
Unverified Commit 1564a81a authored by David Reguera's avatar David Reguera Committed by GitHub
Browse files

16059 - Add missing type hints for ASTModel (#25364)



* 16059 - Add missing type hints for ASTModel

* Add an additional type hint
Co-authored-by: default avatarMatt <Rocketknight1@users.noreply.github.com>

---------
Co-authored-by: default avatarMatt <Rocketknight1@users.noreply.github.com>
parent 1367142a
......@@ -443,7 +443,7 @@ AUDIO_SPECTROGRAM_TRANSFORMER_INPUTS_DOCSTRING = r"""
AUDIO_SPECTROGRAM_TRANSFORMER_START_DOCSTRING,
)
class ASTModel(ASTPreTrainedModel):
def __init__(self, config: ASTConfig):
def __init__(self, config: ASTConfig) -> None:
super().__init__(config)
self.config = config
......@@ -481,7 +481,7 @@ class ASTModel(ASTPreTrainedModel):
output_attentions: Optional[bool] = None,
output_hidden_states: Optional[bool] = None,
return_dict: Optional[bool] = None,
):
) -> Union[Tuple, BaseModelOutputWithPooling]:
output_attentions = output_attentions if output_attentions is not None else self.config.output_attentions
output_hidden_states = (
output_hidden_states if output_hidden_states is not None else self.config.output_hidden_states
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment