From 4d7704388696113c56779b4d7a43da5e9523f522 Mon Sep 17 00:00:00 2001 From: Awni Hannun Date: Wed, 5 Mar 2025 13:53:35 -0800 Subject: [PATCH] fix flaky test --- llms/tests/test_prompt_cache.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/llms/tests/test_prompt_cache.py b/llms/tests/test_prompt_cache.py index de5694d5..c1860892 100644 --- a/llms/tests/test_prompt_cache.py +++ b/llms/tests/test_prompt_cache.py @@ -298,7 +298,7 @@ class TestPromptCache(unittest.TestCase): ): i += 1 self.assertEqual(tok, toks[i]) - self.assertTrue(mx.allclose(logits, all_logits[i], rtol=2e-2)) + self.assertTrue(mx.allclose(logits, all_logits[i], rtol=3e-2)) if __name__ == "__main__":