From 9694e9c50239fb7a36cc08e2b9c4184f7af12c76 Mon Sep 17 00:00:00 2001 From: Lorenzo Stella Date: Thu, 17 Oct 2024 13:35:57 +0200 Subject: [PATCH] try forcing float64 --- test/torch/distribution/test_negative_binomial.py | 5 +---- 1 file changed, 1 insertion(+), 4 deletions(-) diff --git a/test/torch/distribution/test_negative_binomial.py b/test/torch/distribution/test_negative_binomial.py index 4c53172e28..0b7e9d7cc7 100644 --- a/test/torch/distribution/test_negative_binomial.py +++ b/test/torch/distribution/test_negative_binomial.py @@ -47,9 +47,6 @@ def test_custom_neg_bin_cdf(total_count, probs, value): assert np.allclose(torch_cdf, scipy_cdf) -@pytest.mark.skipif( - sys.version_info.major == 3 and sys.version_info.minor == 9, reason="test fails on python 3.9" -) @pytest.mark.parametrize("probs", [0.1, 0.5, 0.8]) @pytest.mark.parametrize("total_count", [3, 7, 100]) @pytest.mark.parametrize("value", [0.1, 0.5, 0.9]) @@ -57,7 +54,7 @@ def test_custom_neg_bin_icdf(total_count, probs, value): torch_dist = NegativeBinomial(total_count=total_count, probs=probs) scipy_dist = torch_dist.scipy_nbinom - torch_icdf = torch_dist.icdf(torch.as_tensor(value)).numpy() + torch_icdf = torch_dist.icdf(torch.as_tensor(value, dtype=torch.float64)).numpy() scipy_icdf = scipy_dist.ppf(np.asarray(value)) assert np.allclose(torch_icdf, scipy_icdf)