Fix progressbar with nested compound step samplers

.github/workflows/tests.yml

@@ -69,6 +69,7 @@ jobs:
             tests/distributions/test_shape_utils.py
             tests/distributions/test_mixture.py
             tests/test_testing.py
+            tests/test_progress_bar.py
 
           - |
             tests/distributions/test_continuous.py

pymc/backends/arviz.py

@@ -39,8 +39,9 @@
 import pymc
 
 from pymc.model import Model, modelcontext
+from pymc.progress_bar import CustomProgress, default_progress_theme
 from pymc.pytensorf import PointFunc, extract_obs_data
-from pymc.util import CustomProgress, default_progress_theme, get_default_varnames
+from pymc.util import get_default_varnames
 
 if TYPE_CHECKING:
     from pymc.backends.base import MultiTrace

pymc/logprob/abstract.py

@@ -35,6 +35,7 @@
 #   SOFTWARE.
 
 import abc
+import warnings
 
 from collections.abc import Sequence
 from functools import singledispatch