set mypy version to 1.9.0

Author: jsbaan

.github/workflows/mypy-type-checking.yml

@@ -25,7 +25,7 @@ jobs:
     - name: Install dependencies
       run: |
         python -m pip install --upgrade pip
-        pip install mypy
+        pip install mypy==1.9.0
         if [ -f requirements.txt ]; then pip install -r requirements.txt; fi
     - name: Type checking with mypy
       run: |

encoder.py

@@ -1,5 +1,6 @@
 import unittest
 import math
+from typing import Optional
 
 import torch
 from torch import nn
@@ -38,7 +39,7 @@ def _reset_parameters(self):
                 xavier_uniform_(p)
 
     def forward(
-        self, input_ids: torch.Tensor, src_padding_mask: torch.BoolTensor = None
+        self, input_ids: torch.Tensor, src_padding_mask: Optional[torch.BoolTensor] = None
     ):
         """
         Performs one encoder forward pass given input token ids and an optional attention mask.
@@ -72,7 +73,7 @@ def __init__(self, hidden_dim: int, ff_dim: int, num_heads: int, dropout_p: floa
         self.layer_norm1 = nn.LayerNorm(hidden_dim)
         self.layer_norm2 = nn.LayerNorm(hidden_dim)
 
-    def forward(self, x: torch.FloatTensor, src_padding_mask: torch.BoolTensor = None):
+    def forward(self, x: torch.FloatTensor, src_padding_mask: Optional[torch.BoolTensor] = None):
         """
         Performs one encoder *block* forward pass given the previous block's output and an optional attention mask.