Skip to content

Commit a02b37d

Browse files
algoriddlefacebook-github-bot
authored andcommitted
relax test_lut rtol (facebookresearch#3016)
Summary: Pull Request resolved: facebookresearch#3016 To resolve test failure for raft nightly. Reviewed By: mdouze Differential Revision: D48465199 fbshipit-source-id: 6bfa3c585e6c3d540e1f58a6351e61d96a54e0c0
1 parent 69cb877 commit a02b37d

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

tests/test_local_search_quantizer.py

+1-2
Original file line numberDiff line numberDiff line change
@@ -581,8 +581,7 @@ def test_lut(self):
581581
lut_ref[:, i] = xq[:, i] @ codebooks[i].T
582582
lut_ref = lut_ref.reshape(nq, codebook_size)
583583

584-
# max rtoal in OSX: 2.87e-6
585-
np.testing.assert_allclose(lut, lut_ref, rtol=1e-04)
584+
np.testing.assert_allclose(lut, lut_ref, rtol=5e-04)
586585

587586

588587
class TestIndexProductLocalSearchQuantizer(unittest.TestCase):

0 commit comments

Comments
 (0)