Remove PolyaGamma pin and add to conda env

Author: ricardoV94

.github/workflows/mypy.yml

@@ -26,7 +26,6 @@ jobs:
       - name: Install-pymc and mypy dependencies
         run: |
           pip install -e .
-          pip install --pre -U polyagamma
           python --version
       - name: Run mypy
         run: |

.github/workflows/tests.yml

@@ -158,8 +158,6 @@ jobs:
       - name: Install-pymc
         run: |
           pip install -e .
-          # TODO: https://github.com/pymc-devs/pymc/issues/7417
-          pip install --pre -U 'polyagamma<1.3.7'
           python --version
           micromamba list
       - name: Run tests
@@ -210,7 +208,6 @@ jobs:
       - name: Install-pymc
         run: |
           pip install -e .
-          pip install --pre -U 'polyagamma<1.3.7'
           python --version
           micromamba list
       - name: Run tests
@@ -363,7 +360,6 @@ jobs:
       - name: Install-pymc
         run: |
           pip install -e .
-          pip install --pre -U 'polyagamma<1.3.7'
           python --version
           micromamba list
       - name: Run tests

conda-envs/environment-dev.yml

@@ -26,6 +26,7 @@ dependencies:
 - myst-nb<=1.0.0
 - numpydoc
 - pre-commit>=2.8.0
+- polyagamma
 - pytest-cov>=2.5
 - pytest>=3.0
 - rich>=13.7.1
@@ -35,7 +36,6 @@ dependencies:
 - sphinx>=1.5
 - sphinxext-rediraffe
 - watermark
-- polyagamma
 - sphinx-remove-toctrees
 - mypy=1.5.1
 - types-cachetools

conda-envs/environment-test.yml

@@ -13,6 +13,7 @@ dependencies:
 - numpy>=1.25.0
 - pandas>=0.24.0
 - pip
+- polyagamma
 - pytensor>=2.26.2,<2.28
 - python-graphviz
 - networkx

conda-envs/windows-environment-dev.yml

@@ -24,6 +24,7 @@ dependencies:
 - ipython>=7.16
 - myst-nb<=1.0.0
 - numpydoc
+- polyagamma
 - pre-commit>=2.8.0
 - pytest-cov>=2.5
 - pytest>=3.0

conda-envs/windows-environment-test.yml

@@ -14,6 +14,7 @@ dependencies:
 - numpy>=1.25.0
 - pandas>=0.24.0
 - pip
+- polyagamma
 - pytensor>=2.26.2,<2.28
 - python-graphviz
 - networkx

tests/distributions/test_continuous.py

@@ -2440,7 +2440,9 @@ def test_rng_different_shapes(self):
 )
 class TestPolyaGamma(BaseTestDistributionRandom):
     def polyagamma_rng_fn(self, size, h, z, rng):
-        return random_polyagamma(h, z, size=size, random_state=rng._bit_generator)
+        # Polyagamma returns different values if inputs have explicit broadcasted dims
+        # Which PyTensor RVs always do when size is not None.
+        return random_polyagamma(np.atleast_1d(h), np.atleast_1d(z), size=size, random_state=rng)
 
     pymc_dist = pm.PolyaGamma
     pymc_dist_params = {"h": 1.0, "z": 0.0}