fix test failures for float32

pymc-devs · Jul 10, 2021 · 0aae7ce · 0aae7ce
1 parent 7b8d33b
commit 0aae7ce
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/.github/workflows/pytest.yml b/.github/workflows/pytest.yml
@@ -134,7 +134,7 @@ jobs:
         run: |
           conda activate pymc3-dev-py37
           pip install -e .
-          pip install --pre -U polyagamma>=1.3.2b2
+          pip install --pre -U polyagamma
           python --version
       - name: Run tests
         run: |

diff --git a/pymc3/distributions/continuous.py b/pymc3/distributions/continuous.py
@@ -4115,7 +4115,7 @@ def perform(self, node, ins, outs):
             polyagamma_pdf(x, h, z, return_log=True)
             if self.get_pdf
             else polyagamma_cdf(x, h, z, return_log=True)
-        )
+        ).astype(aesara.config.floatX)
 
 
 class PolyaGamma(PositiveContinuous):