Fixed typo in the test

pytorch · Mar 7, 2023 · 5143a0f · 5143a0f
1 parent 5aaafe1
commit 5143a0f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/test/distributions/test_distributions.py b/test/distributions/test_distributions.py
@@ -2600,7 +2600,7 @@ def test_gumbel(self):
                                 validate_args=False).cdf(-5.0), 0.0, atol=1e-4, rtol=0)
         self.assertEqual(Gumbel(torch.tensor(0.0, dtype=torch.float64),
                                 torch.tensor(1.0, dtype=torch.float64),
-                                validate_args=False).cdf(-10.0), 1.0, atol=1e-4, rtol=0)
+                                validate_args=False).cdf(-10.0), 0.0, atol=1e-8, rtol=0)
 
         def ref_log_prob(idx, x, log_prob):
             l = loc.view(-1)[idx].detach()