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""" ...@@ -443,7 +443,7 @@ AUDIO_SPECTROGRAM_TRANSFORMER_INPUTS_DOCSTRING = r"""
AUDIO_SPECTROGRAM_TRANSFORMER_START_DOCSTRING, AUDIO_SPECTROGRAM_TRANSFORMER_START_DOCSTRING,
) )
class ASTModel(ASTPreTrainedModel): class ASTModel(ASTPreTrainedModel):
def __init__(self, config: ASTConfig): def __init__(self, config: ASTConfig) -> None:
super().__init__(config) super().__init__(config)
self.config = config self.config = config
...@@ -481,7 +481,7 @@ class ASTModel(ASTPreTrainedModel): ...@@ -481,7 +481,7 @@ class ASTModel(ASTPreTrainedModel):
output_attentions: Optional[bool] = None, output_attentions: Optional[bool] = None,
output_hidden_states: Optional[bool] = None, output_hidden_states: Optional[bool] = None,
return_dict: 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_attentions = output_attentions if output_attentions is not None else self.config.output_attentions
output_hidden_states = ( output_hidden_states = (
output_hidden_states if output_hidden_states is not None else self.config.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