Transferred imports of rcParams and requires from arviz to arviz_base

Author: MSK-005

conda-envs/environment-dev.yml

@@ -6,7 +6,7 @@ channels:
 dependencies:
 # Base dependencies
 - arviz>=0.13.0
-- arviz-base
+- arviz-base>=0.7.0
 - blas
 - cachetools>=4.2.1
 - cloudpickle

conda-envs/environment-docs.yml

@@ -6,7 +6,7 @@ channels:
 dependencies:
 # Base dependencies
 - arviz>=0.13.0
-- arviz-base
+- arviz-base>=0.7.0
 - cachetools>=4.2.1
 - cloudpickle
 - numpy>=1.25.0

conda-envs/environment-test.yml

@@ -6,6 +6,7 @@ channels:
 dependencies:
 # Base dependencies
 - arviz>=0.13.0
+- arviz_base>=0.7.0
 - blas
 - cachetools>=4.2.1
 - cloudpickle

pymc/backends/arviz.py

@@ -28,9 +28,11 @@
 import numpy as np
 import xarray
 
-from arviz import InferenceData, concat, rcParams
-from arviz.data.base import CoordSpec, DimSpec, requires
+from arviz import InferenceData, concat
+from arviz.data.base import CoordSpec, DimSpec
 from arviz_base import dict_to_dataset
+from arviz_base.base import requires
+from arviz_base.rcparams import RcParams
 from pytensor.graph import ancestors
 from pytensor.tensor.sharedvar import SharedVariable
 from rich.progress import Console
@@ -212,7 +214,7 @@ def __init__(
         save_warmup: bool | None = None,
         include_transformed: bool = False,
     ):
-        self.save_warmup = rcParams["data.save_warmup"] if save_warmup is None else save_warmup
+        self.save_warmup = RcParams["data.save_warmup"] if save_warmup is None else save_warmup
         self.include_transformed = include_transformed
         self.trace = trace
 

requirements-dev.txt

@@ -1,7 +1,7 @@
 # This file is auto-generated by scripts/generate_pip_deps_from_conda.py, do not modify.
 # See that file for comments about the need/usage of each dependency.
 
-arviz-base
+arviz-base>=0.7.0
 arviz>=0.13.0
 cachetools>=4.2.1
 cloudpickle