From 5cd17c81bd18cd93853a519e87c5d896c6b6cc96 Mon Sep 17 00:00:00 2001 From: Ricardo Vieira Date: Wed, 29 Mar 2023 09:24:22 +0200 Subject: [PATCH] Fix PyMC API change --- conda-envs/environment-test.yml | 2 +- conda-envs/windows-environment-test.yml | 2 +- pymc_experimental/marginal_model.py | 2 +- requirements.txt | 2 +- 4 files changed, 4 insertions(+), 4 deletions(-) diff --git a/conda-envs/environment-test.yml b/conda-envs/environment-test.yml index 6e8b6b484..b034fc9c1 100644 --- a/conda-envs/environment-test.yml +++ b/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 diff --git a/conda-envs/windows-environment-test.yml b/conda-envs/windows-environment-test.yml index e179b7a96..3e3b61f43 100644 --- a/conda-envs/windows-environment-test.yml +++ b/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 diff --git a/pymc_experimental/marginal_model.py b/pymc_experimental/marginal_model.py index 599509bfb..ed40325fb 100644 --- a/pymc_experimental/marginal_model.py +++ b/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 diff --git a/requirements.txt b/requirements.txt index af926a86c..600e3fa57 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1 +1 @@ -pymc>=5.1.2 +pymc>=5.2.0