Fix lint issues.

pytorch · Oct 3, 2022 · f022db6 · f022db6
1 parent 0b0de10
commit f022db6
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/test/test_mps.py b/test/test_mps.py
@@ -4885,10 +4885,10 @@ def helper(probs, compare_mean, compare_var, num_samples=5, replacement=True):
                 print(mps_out.to('cpu').float().std() ** 2, compare_var)
 
         # TODO: Add tests for data types
-        helper(np.array([[0., 0., 0., 0.5, 0.5]]), (3 + 4)/2, (12.5 - 3.5 ** 2), 100000)
-        helper(np.array([[.2, .2, .2, .2, .2]]), (0 + 1 + 2 + 3 + 4)/5, (6 - 2 * 2), 10000)
-        helper(np.array([[1, 1, 1, 1, 1]]), (0 + 1 + 2 + 3 + 4)/5, (6 - 2 * 2), 10000)
-        helper(np.array([1, 1, 1, 1, 1]), (0 + 1 + 2 + 3 + 4)/5, (6 - 2 * 2), 10000)
+        helper(np.array([[0., 0., 0., 0.5, 0.5]]), (3 + 4) / 2, (12.5 - 3.5 ** 2), 100000)
+        helper(np.array([[.2, .2, .2, .2, .2]]), (0 + 1 + 2 + 3 + 4) / 5, (6 - 2 * 2), 10000)
+        helper(np.array([[1, 1, 1, 1, 1]]), (0 + 1 + 2 + 3 + 4) / 5, (6 - 2 * 2), 10000)
+        helper(np.array([1, 1, 1, 1, 1]), (0 + 1 + 2 + 3 + 4) / 5, (6 - 2 * 2), 10000)
         helper(np.array([[1, 1, 1, 1, 1, 1, 1]]), 0, 0, 7, False)
 
 class TestNNMPS(NNTestCase):