From 3c2299aa5b413dda037a1677fcb307d1fc5d8bbb Mon Sep 17 00:00:00 2001 From: austereantelope <95935342+austereantelope@users.noreply.github.com> Date: Thu, 10 Feb 2022 14:58:43 -0600 Subject: [PATCH] tighten test_sampled_equals_unsampled_when_biased_against_non_sampled_positions bound (#5549) Co-authored-by: Akshita Bhagia --- tests/modules/sampled_softmax_loss_test.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/modules/sampled_softmax_loss_test.py b/tests/modules/sampled_softmax_loss_test.py index 489ca299647..b2574dc8e41 100644 --- a/tests/modules/sampled_softmax_loss_test.py +++ b/tests/modules/sampled_softmax_loss_test.py @@ -95,4 +95,4 @@ def fake_choice(num_words: int, num_samples: int) -> Tuple[np.ndarray, int]: # Should be close pct_error = (sampled_loss - full_loss) / full_loss - assert abs(pct_error) < 0.001 + assert abs(pct_error) < 0.0003