Add TorchFix linter

.flake8

@@ -1,9 +1,13 @@
 [flake8]
 ignore =
     E401,E402,E501,E722,W503,W504,F821,B006,B007,B008,B009,
-    E203 # https://github.com/PyCQA/pycodestyle/issues/373
+    # https://github.com/PyCQA/pycodestyle/issues/373
+    E203
 select =
     B,C,E,F,P,T4,W,B9,
-    D417 # Missing argument descriptions in the docstring
+    # Missing argument descriptions in the docstring
+    D417,
+    # TorchFix
+    TOR
 max-line-length = 120
 exclude = docs/source,third_party

.pre-commit-config.yaml

@@ -34,5 +34,6 @@ repos:
       - id: flake8
         additional_dependencies:
           - flake8-docstrings == 1.6.0
+          - torchfix == 0.0.1
         args:
           - --config=.flake8

test/torchtext_unittest/data/test_jit.py

@@ -1,5 +1,5 @@
 import torch
-from torch.testing import assert_allclose
+from torch.testing import assert_close
 from torchtext.nn import InProjContainer, MultiheadAttentionContainer, ScaledDotProduct
 
 from ..common.torchtext_test_case import TorchtextTestCase
@@ -26,5 +26,5 @@ def test_torchscript_multiheadattention(self) -> None:
 
         ts_MHA = torch.jit.script(MHA)
         ts_mha_output, ts_attn_weights = ts_MHA(query, key, value, attn_mask=attn_mask)
-        assert_allclose(mha_output, ts_mha_output)
-        assert_allclose(attn_weights, ts_attn_weights)
+        assert_close(mha_output, ts_mha_output)
+        assert_close(attn_weights, ts_attn_weights)