Fix lint build (#60438)

Mike Ruberry · facebook-github-bot · commit 15dc320cae4e · 2021-06-22T00:11:55.000-07:00
Summary: per title Fixes #{issue number} Pull Request resolved: #60438 Reviewed By: ngimel Differential Revision: D29288175 Pulled By: mruberry fbshipit-source-id: f59b579b1793fdb1d298109c2bef0a70badb37b4
diff --git a/torch/testing/_asserts.py b/torch/testing/_asserts.py
@@ -76,8 +76,7 @@ def wrapper(
             relaxed_complex_nan = False
 
         if actual.dtype not in (torch.complex32, torch.complex64, torch.complex128):
-            return check_tensors(actual, expected, equal_nan=equal_nan, **kwargs)  # type: ignore[call-arg]
-
+            return check_tensors(actual, expected, equal_nan=equal_nan, **kwargs)
         if relaxed_complex_nan:
             actual, expected = [
                 t.clone().masked_fill(
@@ -86,11 +85,11 @@ def wrapper(
                 for t in (actual, expected)
             ]
 
-        error_meta = check_tensors(actual.real, expected.real, equal_nan=equal_nan, **kwargs)  # type: ignore[call-arg]
+        error_meta = check_tensors(actual.real, expected.real, equal_nan=equal_nan, **kwargs)
         if error_meta:
             return error_meta.amend_msg(postfix="\n\nThe failure occurred for the real part.")
 
-        error_meta = check_tensors(actual.imag, expected.imag, equal_nan=equal_nan, **kwargs)  # type: ignore[call-arg]
+        error_meta = check_tensors(actual.imag, expected.imag, equal_nan=equal_nan, **kwargs)
         if error_meta:
             return error_meta.amend_msg(postfix="\n\nThe failure occurred for the imaginary part.")
 
