From d348c96a579ea92bb32d33b47be4bc1a1b5052d3 Mon Sep 17 00:00:00 2001 From: Awni Hannun Date: Wed, 5 Mar 2025 14:00:09 -0800 Subject: [PATCH] fix flaky test (#1322) --- tests/test_prompt_cache.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/test_prompt_cache.py b/tests/test_prompt_cache.py index de5694d..c186089 100644 --- a/tests/test_prompt_cache.py +++ b/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__":