diff --git a/conda-envs/environment-test.yml b/conda-envs/environment-test.yml index 97b8379ae..c0582f832 100644 --- a/conda-envs/environment-test.yml +++ b/conda-envs/environment-test.yml @@ -3,9 +3,10 @@ channels: - conda-forge - nodefaults dependencies: - - pymc>=5.21.1 + - pymc>=5.24.1 + - pytensor>=2.31.4 - scikit-learn - - better-optimize>=0.1.2 + - better-optimize>=0.1.5 - dask<2025.1.1 - xhistogram - statsmodels @@ -13,7 +14,7 @@ dependencies: - pytest - pytest-cov - pydantic>=2.0.0 - - preliz>=0.5.0 + - preliz>=0.20.0 - pip - pip: - jax diff --git a/pymc_extras/inference/pathfinder/pathfinder.py b/pymc_extras/inference/pathfinder/pathfinder.py index 774541bc4..e14fa1b20 100644 --- a/pymc_extras/inference/pathfinder/pathfinder.py +++ b/pymc_extras/inference/pathfinder/pathfinder.py @@ -38,16 +38,15 @@ from pymc.initial_point import make_initial_point_fn from pymc.model import modelcontext from pymc.model.core import Point +from pymc.progress_bar import CustomProgress, default_progress_theme from pymc.pytensorf import ( compile, find_rng_nodes, reseed_rngs, ) from pymc.util import ( - CustomProgress, RandomSeed, _get_seeds_per_chain, - default_progress_theme, get_default_varnames, ) from pytensor.compile.function.types import Function diff --git a/pyproject.toml b/pyproject.toml index c90ff1c4d..c7bc7e325 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -34,10 +34,12 @@ keywords = [ license = {file = "LICENSE"} dynamic = ["version"] # specify the version in the __init__.py file dependencies = [ - "pymc>=5.21.1", + "pymc>=5.24.1", + "pytensor>=2.31.4", "scikit-learn", - "better-optimize>=0.1.4", + "better-optimize>=0.1.5", "pydantic>=2.0.0", + "preliz>=0.20.0", ] [project.optional-dependencies] @@ -51,7 +53,6 @@ dev = [ "dask[all]<2025.1.1", "blackjax", "statsmodels", - "preliz>=0.5.0", ] docs = [ "nbsphinx>=0.4.2",