added type hints (#19076)

0d1ba2dd · Partho · GitHub · 6f25d107 · 0d1ba2dd
Unverified Commit 0d1ba2dd authored Sep 19, 2022 by Partho Committed by GitHub Sep 19, 2022
Show whitespace changes
Inline Side-by-side

Showing with 5 additions and 5 deletions

src/transformers/models/sew_d/modeling_sew_d.py src/transformers/models/sew_d/modeling_sew_d.py +5 -5

No files found.
--- a/src/transformers/models/sew_d/modeling_sew_d.py
+++ b/src/transformers/models/sew_d/modeling_sew_d.py
@@ -1194,11 +1194,11 @@ class SEWDEncoder(nn.Module):
    def forward(
        self,
-        hidden_states,
+        hidden_states: torch.tensor,
-        attention_mask=None,
+        attention_mask: Optional[torch.Tensor] = None,
-        output_attentions=False,
+        output_attentions: bool = False,
-        output_hidden_states=False,
+        output_hidden_states: bool = False,
-        return_dict=True,
+        return_dict: bool = True,
    ):
        max_encoder_length = hidden_states.shape[1] // self.config.squeeze_factor
        if attention_mask is None: