Skip to content

Commit ce85187

Browse files
author
Vahid Tavanashad
committed
update atol for a test
1 parent 69feed8 commit ce85187

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

dpnp/tests/test_linalg.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -2410,7 +2410,9 @@ def test_qr(self, dtype, shape, mode):
24102410
if mode in ("complete", "reduced"):
24112411
result = dpnp.linalg.qr(ia, mode)
24122412
dpnp_q, dpnp_r = result.Q, result.R
2413-
assert dpnp.allclose(dpnp.matmul(dpnp_q, dpnp_r), ia)
2413+
assert dpnp.allclose(
2414+
dpnp.matmul(dpnp_q, dpnp_r), ia, atol=1e-05
2415+
)
24142416
else: # mode=="raw"
24152417
dpnp_q, dpnp_r = dpnp.linalg.qr(ia, mode)
24162418
assert_dtype_allclose(dpnp_q, np_q, factor=24)
@@ -2439,7 +2441,7 @@ def test_qr_large(self, dtype, shape, mode):
24392441
if mode in ("complete", "reduced"):
24402442
result = dpnp.linalg.qr(ia, mode)
24412443
dpnp_q, dpnp_r = result.Q, result.R
2442-
assert dpnp.allclose(dpnp.matmul(dpnp_q, dpnp_r), ia, atol=1e-4)
2444+
assert dpnp.allclose(dpnp.matmul(dpnp_q, dpnp_r), ia, atol=1e-5)
24432445
else: # mode=="raw"
24442446
dpnp_q, dpnp_r = dpnp.linalg.qr(ia, mode)
24452447
assert_allclose(dpnp_q, np_q, atol=1e-4)

0 commit comments

Comments
 (0)