From b0e2924f7171129ac8110d120948b1c5c1f4b4e4 Mon Sep 17 00:00:00 2001
From: Adam Laiacano <108741458+nv-alaiacano@users.noreply.github.com>
Date: Fri, 26 May 2023 16:48:25 -0400
Subject: [PATCH] increase tolerance threshold (#1123)

---
 tests/unit/tf/transformers/test_block.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tests/unit/tf/transformers/test_block.py b/tests/unit/tf/transformers/test_block.py
index 1f97433b33..ffa373e911 100644
--- a/tests/unit/tf/transformers/test_block.py
+++ b/tests/unit/tf/transformers/test_block.py
@@ -81,7 +81,7 @@ def test_retrieval_transformer(sequence_testing_data: Dataset, run_eagerly):
     assert list(item_embeddings.shape) == [101, d_model]
     predicitons_2 = np.dot(query_embeddings, item_embeddings.T)
 
-    np.testing.assert_allclose(predictions, predicitons_2, atol=1e-4)
+    np.testing.assert_allclose(predictions, predicitons_2, atol=1e-3)
 
 
 def test_transformer_encoder():