Add float32 xfail to test_rice

Author: ricardoV94

pymc3/tests/test_distributions.py

@@ -2544,18 +2544,24 @@ def test_multidimensional_beta_construction(self):
         with Model():
             Beta("beta", alpha=1.0, beta=1.0, size=(10, 20))
 
+    @pytest.mark.xfail(
+        condition=(aesara.config.floatX == "float32"),
+        reason="Some combinations underflow to -inf in float32 in pymc version",
+    )
     def test_rice(self):
         self.check_logp(
             Rice,
             Rplus,
-            {"nu": Rplus, "sigma": Rplusbig},
-            lambda value, nu, sigma: sp.rice.logpdf(value, b=nu / sigma, loc=0, scale=sigma),
+            {"b": Rplus, "sigma": Rplusbig},
+            lambda value, b, sigma: sp.rice.logpdf(value, b=b, loc=0, scale=sigma),
         )
+
+    def test_rice_nu(self):
         self.check_logp(
             Rice,
             Rplus,
-            {"b": Rplus, "sigma": Rplusbig},
-            lambda value, b, sigma: sp.rice.logpdf(value, b=b, loc=0, scale=sigma),
+            {"nu": Rplus, "sigma": Rplusbig},
+            lambda value, nu, sigma: sp.rice.logpdf(value, b=nu / sigma, loc=0, scale=sigma),
         )
 
     def test_moyal_logp(self):