Update test import exception

pymc-devs · Mar 28, 2021 · 7cc0746 · 7cc0746
1 parent 3a9a4ce
commit 7cc0746
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pymc3/tests/test_distributions_random.py b/pymc3/tests/test_distributions_random.py
@@ -31,7 +31,7 @@
 try:
     pm.PolyaGamma.rv_op.rng_fn()
     polyagamma_not_installed = False
-except RuntimeError:  # pragma: no cover
+except (AttributeError, RuntimeError):  # pragma: no cover
     polyagamma_not_installed = True
 
 from pymc3.aesaraf import floatX, intX