From a8f699efbeee6985f99a9f754b9c814ac0958a7e Mon Sep 17 00:00:00 2001
From: Gergely Szilvasy <gsz@meta.com>
Date: Fri, 18 Aug 2023 02:14:09 -0700
Subject: [PATCH] relax test_lut rtol

Summary: To resolve test failure for raft nightly.

Differential Revision: D48465199

fbshipit-source-id: 42bc3d905110462a06f66b5aa3e2e85d173a175b
---
 tests/test_local_search_quantizer.py | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/tests/test_local_search_quantizer.py b/tests/test_local_search_quantizer.py
index 4c239527db..01fec70ccf 100644
--- a/tests/test_local_search_quantizer.py
+++ b/tests/test_local_search_quantizer.py
@@ -581,8 +581,7 @@ def test_lut(self):
             lut_ref[:, i] = xq[:, i] @ codebooks[i].T
         lut_ref = lut_ref.reshape(nq, codebook_size)
 
-        # max rtoal in OSX: 2.87e-6
-        np.testing.assert_allclose(lut, lut_ref, rtol=1e-04)
+        np.testing.assert_allclose(lut, lut_ref, rtol=5e-04)
 
 
 class TestIndexProductLocalSearchQuantizer(unittest.TestCase):