Fix assert_allclose warning

michaelbyun · Mar 25, 2023 · ee3911e · ee3911e
1 parent b1b95e5
commit ee3911e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/test_eigsh.py b/tests/test_eigsh.py
@@ -20,7 +20,7 @@ def test_lanczos_eigsh(n, which):
  w_scipy, v_scipy = eigsh(A.numpy(), which=which)
 
  # Check that the eigenvalues match to within the tolerance
- torch.testing.assert_allclose(w, torch.from_numpy(w_scipy), atol=1e-3, rtol=1e-3)
+ torch.testing.assert_close(w, torch.from_numpy(w_scipy), atol=1e-3, rtol=1e-3)
 
  # Normalize the sign of the eigenvectors
  for i in range(v.shape[-1]):
@@ -30,4 +30,4 @@ def test_lanczos_eigsh(n, which):
  v_scipy[:, i] *= -1
 
  # Check that the eigenvectors match to within the tolerance
- torch.testing.assert_allclose(v, torch.from_numpy(v_scipy), atol=1e-3, rtol=1e-3)
+ torch.testing.assert_close(v, torch.from_numpy(v_scipy), atol=1e-3, rtol=1e-3)