reduce rtol in test_multihot_empty_rows

NVIDIA-Merlin · Jun 5, 2023 · 26ce1eb · 26ce1eb
1 parent 0c3d403
commit 26ce1eb
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/unit/framework_utils/test_tf_layers.py b/tests/unit/framework_utils/test_tf_layers.py
@@ -318,4 +318,4 @@ def test_multihot_empty_rows():
     )
 
     y_hat = model(x).numpy()
-    np.testing.assert_allclose(y_hat, multi_hot_embedding_rows, rtol=1e-06)
+    np.testing.assert_allclose(y_hat, multi_hot_embedding_rows, rtol=1e-05)