Add TorchFix linter (#2179)

* Add TorchFix linter

* Move comments to separate lines

* Change assert_allclose to assert_close

.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)