fix import error due to moving progress bar functions in latest pymc (#542)

* import progress bar functions from pymc.progress_bar instead of pymc.util

* pinned pymc version to 5.24.0

* Update version pins and dependencies

* Update pymc verison pin

---------

Co-authored-by: jessegrabowski <[email protected]>

Author: Dekermanjian

conda-envs/environment-test.yml

@@ -3,17 +3,18 @@ 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
   - numba
   - pytest
   - pytest-cov
   - pydantic>=2.0.0
-  - preliz>=0.5.0
+  - preliz>=0.20.0
   - pip
   - pip:
       - jax

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

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",