Fix PyMC API change

Author: ricardoV94

conda-envs/environment-test.yml

@@ -9,5 +9,5 @@ dependencies:
 - dask
 - xhistogram
 - pip:
-  - pymc>=5.1.2  # CI was failing to resolve
+  - pymc>=5.2.0  # CI was failing to resolve
   - blackjax

conda-envs/windows-environment-test.yml

@@ -9,4 +9,4 @@ dependencies:
 - dask
 - xhistogram
 - pip:
-  - pymc>=5.1.2  # CI was failing to resolve
+  - pymc>=5.2.0  # CI was failing to resolve

pymc_experimental/marginal_model.py

@@ -7,7 +7,7 @@
 from pymc.distributions.discrete import Bernoulli, Categorical, DiscreteUniform
 from pymc.distributions.transforms import Chain
 from pymc.logprob.abstract import _get_measurable_outputs, _logprob
-from pymc.logprob.joint_logprob import factorized_joint_logprob
+from pymc.logprob.basic import factorized_joint_logprob
 from pymc.logprob.transforms import IntervalTransform
 from pymc.model import Model
 from pymc.pytensorf import constant_fold, inputvars

requirements.txt

@@ -1 +1 @@
-pymc>=5.1.2
+pymc>=5.2.0