mypy fix

mindee · Oct 5, 2023 · 1cff3ca · 1cff3ca
1 parent 2e30672
commit 1cff3ca
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/doctr/models/modules/transformer/pytorch.py b/doctr/models/modules/transformer/pytorch.py
@@ -37,7 +37,7 @@ def forward(self, x: torch.Tensor) -> torch.Tensor:
         Returns:
             positional embeddings (batch, max_len, d_model)
         """
-        x = x + self.pe[:, : x.size(1)]  # type: ignore
+        x = x + self.pe[:, : x.size(1)]  # type: ignore[index]
         return self.dropout(x)