increase tolerance threshold (#1123)

NVIDIA-Merlin · May 26, 2023 · b0e2924 · b0e2924
1 parent 2b012fc
commit b0e2924
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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():