arviz 0.23.3__py3-none-any.whl → 1.0.0rc0__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (185) hide show
  1. arviz/__init__.py +52 -367
  2. arviz-1.0.0rc0.dist-info/METADATA +182 -0
  3. arviz-1.0.0rc0.dist-info/RECORD +5 -0
  4. {arviz-0.23.3.dist-info → arviz-1.0.0rc0.dist-info}/WHEEL +1 -2
  5. {arviz-0.23.3.dist-info → arviz-1.0.0rc0.dist-info}/licenses/LICENSE +0 -1
  6. arviz/data/__init__.py +0 -55
  7. arviz/data/base.py +0 -596
  8. arviz/data/converters.py +0 -203
  9. arviz/data/datasets.py +0 -161
  10. arviz/data/example_data/code/radon/radon.json +0 -326
  11. arviz/data/example_data/data/centered_eight.nc +0 -0
  12. arviz/data/example_data/data/non_centered_eight.nc +0 -0
  13. arviz/data/example_data/data_local.json +0 -12
  14. arviz/data/example_data/data_remote.json +0 -58
  15. arviz/data/inference_data.py +0 -2386
  16. arviz/data/io_beanmachine.py +0 -112
  17. arviz/data/io_cmdstan.py +0 -1036
  18. arviz/data/io_cmdstanpy.py +0 -1233
  19. arviz/data/io_datatree.py +0 -23
  20. arviz/data/io_dict.py +0 -462
  21. arviz/data/io_emcee.py +0 -317
  22. arviz/data/io_json.py +0 -54
  23. arviz/data/io_netcdf.py +0 -68
  24. arviz/data/io_numpyro.py +0 -497
  25. arviz/data/io_pyjags.py +0 -378
  26. arviz/data/io_pyro.py +0 -333
  27. arviz/data/io_pystan.py +0 -1095
  28. arviz/data/io_zarr.py +0 -46
  29. arviz/data/utils.py +0 -139
  30. arviz/labels.py +0 -210
  31. arviz/plots/__init__.py +0 -61
  32. arviz/plots/autocorrplot.py +0 -171
  33. arviz/plots/backends/__init__.py +0 -223
  34. arviz/plots/backends/bokeh/__init__.py +0 -166
  35. arviz/plots/backends/bokeh/autocorrplot.py +0 -101
  36. arviz/plots/backends/bokeh/bfplot.py +0 -23
  37. arviz/plots/backends/bokeh/bpvplot.py +0 -193
  38. arviz/plots/backends/bokeh/compareplot.py +0 -167
  39. arviz/plots/backends/bokeh/densityplot.py +0 -239
  40. arviz/plots/backends/bokeh/distcomparisonplot.py +0 -23
  41. arviz/plots/backends/bokeh/distplot.py +0 -183
  42. arviz/plots/backends/bokeh/dotplot.py +0 -113
  43. arviz/plots/backends/bokeh/ecdfplot.py +0 -73
  44. arviz/plots/backends/bokeh/elpdplot.py +0 -203
  45. arviz/plots/backends/bokeh/energyplot.py +0 -155
  46. arviz/plots/backends/bokeh/essplot.py +0 -176
  47. arviz/plots/backends/bokeh/forestplot.py +0 -772
  48. arviz/plots/backends/bokeh/hdiplot.py +0 -54
  49. arviz/plots/backends/bokeh/kdeplot.py +0 -268
  50. arviz/plots/backends/bokeh/khatplot.py +0 -163
  51. arviz/plots/backends/bokeh/lmplot.py +0 -185
  52. arviz/plots/backends/bokeh/loopitplot.py +0 -211
  53. arviz/plots/backends/bokeh/mcseplot.py +0 -184
  54. arviz/plots/backends/bokeh/pairplot.py +0 -328
  55. arviz/plots/backends/bokeh/parallelplot.py +0 -81
  56. arviz/plots/backends/bokeh/posteriorplot.py +0 -324
  57. arviz/plots/backends/bokeh/ppcplot.py +0 -379
  58. arviz/plots/backends/bokeh/rankplot.py +0 -149
  59. arviz/plots/backends/bokeh/separationplot.py +0 -107
  60. arviz/plots/backends/bokeh/traceplot.py +0 -436
  61. arviz/plots/backends/bokeh/violinplot.py +0 -164
  62. arviz/plots/backends/matplotlib/__init__.py +0 -124
  63. arviz/plots/backends/matplotlib/autocorrplot.py +0 -72
  64. arviz/plots/backends/matplotlib/bfplot.py +0 -78
  65. arviz/plots/backends/matplotlib/bpvplot.py +0 -177
  66. arviz/plots/backends/matplotlib/compareplot.py +0 -135
  67. arviz/plots/backends/matplotlib/densityplot.py +0 -194
  68. arviz/plots/backends/matplotlib/distcomparisonplot.py +0 -119
  69. arviz/plots/backends/matplotlib/distplot.py +0 -178
  70. arviz/plots/backends/matplotlib/dotplot.py +0 -116
  71. arviz/plots/backends/matplotlib/ecdfplot.py +0 -70
  72. arviz/plots/backends/matplotlib/elpdplot.py +0 -189
  73. arviz/plots/backends/matplotlib/energyplot.py +0 -113
  74. arviz/plots/backends/matplotlib/essplot.py +0 -180
  75. arviz/plots/backends/matplotlib/forestplot.py +0 -656
  76. arviz/plots/backends/matplotlib/hdiplot.py +0 -48
  77. arviz/plots/backends/matplotlib/kdeplot.py +0 -177
  78. arviz/plots/backends/matplotlib/khatplot.py +0 -241
  79. arviz/plots/backends/matplotlib/lmplot.py +0 -149
  80. arviz/plots/backends/matplotlib/loopitplot.py +0 -144
  81. arviz/plots/backends/matplotlib/mcseplot.py +0 -161
  82. arviz/plots/backends/matplotlib/pairplot.py +0 -355
  83. arviz/plots/backends/matplotlib/parallelplot.py +0 -58
  84. arviz/plots/backends/matplotlib/posteriorplot.py +0 -348
  85. arviz/plots/backends/matplotlib/ppcplot.py +0 -478
  86. arviz/plots/backends/matplotlib/rankplot.py +0 -119
  87. arviz/plots/backends/matplotlib/separationplot.py +0 -97
  88. arviz/plots/backends/matplotlib/traceplot.py +0 -526
  89. arviz/plots/backends/matplotlib/tsplot.py +0 -121
  90. arviz/plots/backends/matplotlib/violinplot.py +0 -148
  91. arviz/plots/bfplot.py +0 -128
  92. arviz/plots/bpvplot.py +0 -308
  93. arviz/plots/compareplot.py +0 -177
  94. arviz/plots/densityplot.py +0 -284
  95. arviz/plots/distcomparisonplot.py +0 -197
  96. arviz/plots/distplot.py +0 -233
  97. arviz/plots/dotplot.py +0 -233
  98. arviz/plots/ecdfplot.py +0 -372
  99. arviz/plots/elpdplot.py +0 -174
  100. arviz/plots/energyplot.py +0 -147
  101. arviz/plots/essplot.py +0 -319
  102. arviz/plots/forestplot.py +0 -304
  103. arviz/plots/hdiplot.py +0 -211
  104. arviz/plots/kdeplot.py +0 -357
  105. arviz/plots/khatplot.py +0 -236
  106. arviz/plots/lmplot.py +0 -380
  107. arviz/plots/loopitplot.py +0 -224
  108. arviz/plots/mcseplot.py +0 -194
  109. arviz/plots/pairplot.py +0 -281
  110. arviz/plots/parallelplot.py +0 -204
  111. arviz/plots/plot_utils.py +0 -599
  112. arviz/plots/posteriorplot.py +0 -298
  113. arviz/plots/ppcplot.py +0 -369
  114. arviz/plots/rankplot.py +0 -232
  115. arviz/plots/separationplot.py +0 -167
  116. arviz/plots/styles/arviz-bluish.mplstyle +0 -1
  117. arviz/plots/styles/arviz-brownish.mplstyle +0 -1
  118. arviz/plots/styles/arviz-colors.mplstyle +0 -2
  119. arviz/plots/styles/arviz-cyanish.mplstyle +0 -1
  120. arviz/plots/styles/arviz-darkgrid.mplstyle +0 -40
  121. arviz/plots/styles/arviz-doc.mplstyle +0 -88
  122. arviz/plots/styles/arviz-docgrid.mplstyle +0 -88
  123. arviz/plots/styles/arviz-grayscale.mplstyle +0 -41
  124. arviz/plots/styles/arviz-greenish.mplstyle +0 -1
  125. arviz/plots/styles/arviz-orangish.mplstyle +0 -1
  126. arviz/plots/styles/arviz-plasmish.mplstyle +0 -1
  127. arviz/plots/styles/arviz-purplish.mplstyle +0 -1
  128. arviz/plots/styles/arviz-redish.mplstyle +0 -1
  129. arviz/plots/styles/arviz-royish.mplstyle +0 -1
  130. arviz/plots/styles/arviz-viridish.mplstyle +0 -1
  131. arviz/plots/styles/arviz-white.mplstyle +0 -40
  132. arviz/plots/styles/arviz-whitegrid.mplstyle +0 -40
  133. arviz/plots/traceplot.py +0 -273
  134. arviz/plots/tsplot.py +0 -440
  135. arviz/plots/violinplot.py +0 -192
  136. arviz/preview.py +0 -58
  137. arviz/py.typed +0 -0
  138. arviz/rcparams.py +0 -606
  139. arviz/sel_utils.py +0 -223
  140. arviz/static/css/style.css +0 -340
  141. arviz/static/html/icons-svg-inline.html +0 -15
  142. arviz/stats/__init__.py +0 -37
  143. arviz/stats/density_utils.py +0 -1013
  144. arviz/stats/diagnostics.py +0 -1013
  145. arviz/stats/ecdf_utils.py +0 -324
  146. arviz/stats/stats.py +0 -2422
  147. arviz/stats/stats_refitting.py +0 -119
  148. arviz/stats/stats_utils.py +0 -609
  149. arviz/tests/__init__.py +0 -1
  150. arviz/tests/base_tests/__init__.py +0 -1
  151. arviz/tests/base_tests/test_data.py +0 -1679
  152. arviz/tests/base_tests/test_data_zarr.py +0 -143
  153. arviz/tests/base_tests/test_diagnostics.py +0 -511
  154. arviz/tests/base_tests/test_diagnostics_numba.py +0 -87
  155. arviz/tests/base_tests/test_helpers.py +0 -18
  156. arviz/tests/base_tests/test_labels.py +0 -69
  157. arviz/tests/base_tests/test_plot_utils.py +0 -342
  158. arviz/tests/base_tests/test_plots_bokeh.py +0 -1288
  159. arviz/tests/base_tests/test_plots_matplotlib.py +0 -2197
  160. arviz/tests/base_tests/test_rcparams.py +0 -317
  161. arviz/tests/base_tests/test_stats.py +0 -925
  162. arviz/tests/base_tests/test_stats_ecdf_utils.py +0 -166
  163. arviz/tests/base_tests/test_stats_numba.py +0 -45
  164. arviz/tests/base_tests/test_stats_utils.py +0 -384
  165. arviz/tests/base_tests/test_utils.py +0 -376
  166. arviz/tests/base_tests/test_utils_numba.py +0 -87
  167. arviz/tests/conftest.py +0 -46
  168. arviz/tests/external_tests/__init__.py +0 -1
  169. arviz/tests/external_tests/test_data_beanmachine.py +0 -78
  170. arviz/tests/external_tests/test_data_cmdstan.py +0 -398
  171. arviz/tests/external_tests/test_data_cmdstanpy.py +0 -496
  172. arviz/tests/external_tests/test_data_emcee.py +0 -166
  173. arviz/tests/external_tests/test_data_numpyro.py +0 -434
  174. arviz/tests/external_tests/test_data_pyjags.py +0 -119
  175. arviz/tests/external_tests/test_data_pyro.py +0 -260
  176. arviz/tests/external_tests/test_data_pystan.py +0 -307
  177. arviz/tests/helpers.py +0 -677
  178. arviz/utils.py +0 -773
  179. arviz/wrappers/__init__.py +0 -13
  180. arviz/wrappers/base.py +0 -236
  181. arviz/wrappers/wrap_pymc.py +0 -36
  182. arviz/wrappers/wrap_stan.py +0 -148
  183. arviz-0.23.3.dist-info/METADATA +0 -264
  184. arviz-0.23.3.dist-info/RECORD +0 -183
  185. arviz-0.23.3.dist-info/top_level.txt +0 -1
arviz/__init__.py CHANGED
@@ -1,386 +1,71 @@
1
- # pylint: disable=wildcard-import,invalid-name,wrong-import-position
2
- """ArviZ is a library for exploratory analysis of Bayesian models."""
3
- __version__ = "0.23.3"
1
+ # pylint: disable=unused-import,unused-wildcard-import,wildcard-import,invalid-name
2
+ """Expose features from _ArviZverse_ refactored packages together in the ``arviz`` namespace."""
4
3
 
5
4
  import logging
6
- import os
7
5
  import re
8
6
 
9
- from matplotlib.colors import LinearSegmentedColormap
10
- from matplotlib.pyplot import style
11
- import matplotlib as mpl
12
- from packaging import version
7
+ _log = logging.getLogger(__name__)
13
8
 
9
+ info = ""
14
10
 
15
- def _warn_once_per_day():
16
- from .preview import info
17
-
18
- # skip warning if all 3 arviz subpackages are already installed
19
- pat = re.compile(r"arviz_(base|stats|plots) available")
20
- if len(pat.findall(info)) == 3:
21
- return
22
-
23
- import datetime
24
- from warnings import warn
25
- from pathlib import Path
26
- from platformdirs import user_cache_dir
27
-
28
- def _atomic_write_text(path: Path, text: str) -> None:
29
- """
30
- Write text to a file atomically.
31
-
32
- Write to a temporary file in the same directory and replace the target.
33
- """
34
- tmp = path.with_suffix(path.suffix + ".tmp")
35
- tmp.write_text(text)
36
- tmp.replace(path)
37
-
38
- warning_dir = Path(user_cache_dir("arviz", "arviz"))
39
- warning_dir.mkdir(exist_ok=True)
40
-
41
- stamp_file = warning_dir / "daily_warning"
42
- today = datetime.date.today()
43
-
44
- try:
45
- last_date = datetime.date.fromisoformat(stamp_file.read_text().strip())
46
- except (FileNotFoundError, ValueError):
47
- last_date = None
48
-
49
- if last_date != today:
50
- warn(
51
- "\nArviZ is undergoing a major refactor to improve flexibility and extensibility "
52
- "while maintaining a user-friendly interface."
53
- "\nSome upcoming changes may be backward incompatible."
54
- "\nFor details and migration guidance, visit: "
55
- "https://python.arviz.org/en/latest/user_guide/migration_guide.html",
56
- FutureWarning,
57
- )
58
- _atomic_write_text(stamp_file, today.isoformat())
59
-
60
-
61
- _warn_once_per_day()
62
-
63
-
64
- class Logger(logging.Logger):
65
- """Override Logger to avoid repeated messages."""
66
-
67
- def __init__(self, name, level=logging.NOTSET):
68
- super().__init__(name=name, level=level)
69
- self.cache = []
70
-
71
- def _log(
72
- self, level, msg, *args, **kwargs
73
- ): # pylint: disable=signature-differs, arguments-differ
74
- msg_hash = hash(msg)
75
- if msg_hash in self.cache:
76
- return
77
- self.cache.append(msg_hash)
78
- super()._log(level, msg, *args, **kwargs)
79
-
80
-
81
- _log = Logger("arviz")
82
-
11
+ try:
12
+ from arviz_base import *
13
+ import arviz_base as base
83
14
 
84
- from .data import *
85
- from .plots import *
86
- from .plots.backends import *
87
- from .stats import *
88
- from .rcparams import rc_context, rcParams
89
- from .utils import Numba, Dask, interactive_backend
90
- from .wrappers import *
91
- from . import preview
15
+ _status = (
16
+ f"arviz_base {base.__version__} available, "
17
+ "exposing its functions as part of the `arviz` namespace"
18
+ )
19
+ _log.info(_status)
20
+ del base
21
+ except ModuleNotFoundError as err:
22
+ raise ImportError("arviz's dependency arviz_base is not installed", name="arviz") from err
92
23
 
93
- # add ArviZ's styles to matplotlib's styles
94
- _arviz_style_path = os.path.join(os.path.dirname(__file__), "plots", "styles")
95
- if version.parse(mpl.__version__) >= version.parse("3.11.0.dev0"):
96
- style.USER_LIBRARY_PATHS.append(_arviz_style_path)
97
- style.reload_library()
98
- else:
99
- style.core.USER_LIBRARY_PATHS.append(_arviz_style_path)
100
- style.core.reload_library()
24
+ info += _status + "\n"
101
25
 
26
+ try:
27
+ from arviz_stats import *
28
+ import arviz_stats as stats
102
29
 
103
- if not logging.root.handlers:
104
- _handler = logging.StreamHandler()
105
- _formatter = logging.Formatter("%(name)s - %(levelname)s - %(message)s")
106
- _handler.setFormatter(_formatter)
107
- _log.setLevel(logging.INFO)
108
- _log.addHandler(_handler)
30
+ # TODO: remove patch. 0.7 version of arviz-stats didn't expose the __version__ attribute
31
+ _status = (
32
+ f"arviz_stats {getattr(stats, '__version__', '0.7.0')} available, "
33
+ "exposing its functions as part of the `arviz` namespace"
34
+ )
35
+ _log.info(_status)
36
+ del stats
37
+ except ModuleNotFoundError as err:
38
+ raise ImportError("arviz's dependency arviz_stats is not installed", name="arviz") from err
109
39
 
40
+ info += _status + "\n"
110
41
 
111
- # adds perceptually uniform grey scale from colorcet
112
- _linear_grey_10_95_c0 = [
113
- [0.10767, 0.1077, 0.1077],
114
- [0.11032, 0.11035, 0.11035],
115
- [0.11295, 0.11298, 0.11297],
116
- [0.11554, 0.11558, 0.11557],
117
- [0.1182, 0.11824, 0.11823],
118
- [0.12079, 0.12083, 0.12082],
119
- [0.12344, 0.12348, 0.12347],
120
- [0.12615, 0.12618, 0.12618],
121
- [0.12879, 0.12882, 0.12881],
122
- [0.13149, 0.13152, 0.13151],
123
- [0.13418, 0.13421, 0.1342],
124
- [0.13684, 0.13688, 0.13687],
125
- [0.13951, 0.13955, 0.13954],
126
- [0.14226, 0.1423, 0.14229],
127
- [0.14499, 0.14503, 0.14502],
128
- [0.1477, 0.14774, 0.14773],
129
- [0.15042, 0.15046, 0.15045],
130
- [0.15313, 0.15317, 0.15316],
131
- [0.15591, 0.15595, 0.15594],
132
- [0.15866, 0.1587, 0.15869],
133
- [0.16142, 0.16147, 0.16145],
134
- [0.16418, 0.16423, 0.16422],
135
- [0.16695, 0.16699, 0.16698],
136
- [0.16973, 0.16977, 0.16976],
137
- [0.17248, 0.17253, 0.17252],
138
- [0.17529, 0.17533, 0.17532],
139
- [0.17811, 0.17815, 0.17814],
140
- [0.18087, 0.18092, 0.1809],
141
- [0.18369, 0.18374, 0.18372],
142
- [0.18652, 0.18656, 0.18655],
143
- [0.18934, 0.18939, 0.18938],
144
- [0.19217, 0.19221, 0.1922],
145
- [0.19502, 0.19506, 0.19505],
146
- [0.19785, 0.1979, 0.19788],
147
- [0.20068, 0.20073, 0.20072],
148
- [0.20357, 0.20362, 0.20361],
149
- [0.20645, 0.2065, 0.20649],
150
- [0.20929, 0.20934, 0.20933],
151
- [0.21219, 0.21224, 0.21222],
152
- [0.21504, 0.21509, 0.21508],
153
- [0.21795, 0.218, 0.21799],
154
- [0.22086, 0.22091, 0.2209],
155
- [0.22374, 0.22379, 0.22377],
156
- [0.22666, 0.22671, 0.22669],
157
- [0.22954, 0.2296, 0.22958],
158
- [0.23248, 0.23253, 0.23252],
159
- [0.23542, 0.23547, 0.23546],
160
- [0.23832, 0.23838, 0.23836],
161
- [0.24127, 0.24133, 0.24131],
162
- [0.24419, 0.24425, 0.24424],
163
- [0.24716, 0.24722, 0.2472],
164
- [0.25009, 0.25015, 0.25014],
165
- [0.25308, 0.25313, 0.25312],
166
- [0.25603, 0.25608, 0.25607],
167
- [0.25902, 0.25908, 0.25906],
168
- [0.26198, 0.26204, 0.26203],
169
- [0.26496, 0.26502, 0.265],
170
- [0.26794, 0.268, 0.26798],
171
- [0.27095, 0.27101, 0.271],
172
- [0.27395, 0.27401, 0.274],
173
- [0.27695, 0.27702, 0.277],
174
- [0.27996, 0.28002, 0.28],
175
- [0.28298, 0.28304, 0.28303],
176
- [0.28598, 0.28604, 0.28603],
177
- [0.28902, 0.28909, 0.28907],
178
- [0.29205, 0.29212, 0.2921],
179
- [0.29508, 0.29514, 0.29513],
180
- [0.29812, 0.29818, 0.29817],
181
- [0.30116, 0.30123, 0.30121],
182
- [0.30422, 0.30429, 0.30427],
183
- [0.30728, 0.30734, 0.30733],
184
- [0.31036, 0.31043, 0.31041],
185
- [0.31342, 0.31349, 0.31347],
186
- [0.31649, 0.31656, 0.31654],
187
- [0.31957, 0.31964, 0.31962],
188
- [0.32266, 0.32273, 0.32271],
189
- [0.32572, 0.3258, 0.32578],
190
- [0.32883, 0.32891, 0.32889],
191
- [0.33193, 0.332, 0.33198],
192
- [0.33504, 0.33512, 0.3351],
193
- [0.33813, 0.3382, 0.33818],
194
- [0.34125, 0.34133, 0.34131],
195
- [0.34436, 0.34444, 0.34442],
196
- [0.3475, 0.34757, 0.34755],
197
- [0.35063, 0.3507, 0.35068],
198
- [0.35374, 0.35382, 0.3538],
199
- [0.35689, 0.35697, 0.35695],
200
- [0.36002, 0.3601, 0.36008],
201
- [0.36317, 0.36325, 0.36323],
202
- [0.36633, 0.36641, 0.36639],
203
- [0.36948, 0.36956, 0.36954],
204
- [0.37263, 0.37272, 0.3727],
205
- [0.3758, 0.37589, 0.37587],
206
- [0.37897, 0.37906, 0.37904],
207
- [0.38214, 0.38223, 0.38221],
208
- [0.38532, 0.3854, 0.38538],
209
- [0.38852, 0.3886, 0.38858],
210
- [0.3917, 0.39179, 0.39177],
211
- [0.39489, 0.39498, 0.39496],
212
- [0.3981, 0.39818, 0.39816],
213
- [0.4013, 0.40138, 0.40136],
214
- [0.40449, 0.40458, 0.40456],
215
- [0.40771, 0.4078, 0.40778],
216
- [0.41093, 0.41102, 0.411],
217
- [0.41415, 0.41423, 0.41421],
218
- [0.41738, 0.41747, 0.41744],
219
- [0.4206, 0.42068, 0.42066],
220
- [0.42383, 0.42392, 0.4239],
221
- [0.42708, 0.42717, 0.42715],
222
- [0.43031, 0.43041, 0.43038],
223
- [0.43355, 0.43364, 0.43362],
224
- [0.43681, 0.43691, 0.43688],
225
- [0.44007, 0.44016, 0.44014],
226
- [0.44333, 0.44342, 0.4434],
227
- [0.44659, 0.44668, 0.44666],
228
- [0.44986, 0.44995, 0.44993],
229
- [0.45313, 0.45322, 0.4532],
230
- [0.4564, 0.4565, 0.45647],
231
- [0.45968, 0.45978, 0.45976],
232
- [0.46296, 0.46306, 0.46303],
233
- [0.46625, 0.46635, 0.46633],
234
- [0.46956, 0.46966, 0.46963],
235
- [0.47284, 0.47294, 0.47292],
236
- [0.47615, 0.47625, 0.47623],
237
- [0.47946, 0.47956, 0.47953],
238
- [0.48276, 0.48286, 0.48284],
239
- [0.48607, 0.48618, 0.48615],
240
- [0.48939, 0.48949, 0.48947],
241
- [0.49271, 0.49281, 0.49279],
242
- [0.49603, 0.49614, 0.49611],
243
- [0.49936, 0.49947, 0.49944],
244
- [0.5027, 0.5028, 0.50278],
245
- [0.50603, 0.50614, 0.50612],
246
- [0.50938, 0.50949, 0.50946],
247
- [0.51273, 0.51284, 0.51281],
248
- [0.51607, 0.51618, 0.51615],
249
- [0.51943, 0.51954, 0.51951],
250
- [0.52279, 0.5229, 0.52287],
251
- [0.52615, 0.52626, 0.52624],
252
- [0.52952, 0.52963, 0.5296],
253
- [0.53289, 0.533, 0.53297],
254
- [0.53626, 0.53637, 0.53634],
255
- [0.53963, 0.53974, 0.53971],
256
- [0.54302, 0.54313, 0.5431],
257
- [0.54641, 0.54652, 0.54649],
258
- [0.54979, 0.5499, 0.54987],
259
- [0.55317, 0.55329, 0.55326],
260
- [0.55657, 0.55669, 0.55666],
261
- [0.55998, 0.56009, 0.56007],
262
- [0.56338, 0.5635, 0.56347],
263
- [0.56679, 0.56691, 0.56688],
264
- [0.57019, 0.57031, 0.57028],
265
- [0.57361, 0.57373, 0.5737],
266
- [0.57703, 0.57715, 0.57712],
267
- [0.58045, 0.58057, 0.58054],
268
- [0.58387, 0.58399, 0.58396],
269
- [0.5873, 0.58742, 0.58739],
270
- [0.59073, 0.59086, 0.59083],
271
- [0.59417, 0.5943, 0.59427],
272
- [0.59761, 0.59774, 0.59771],
273
- [0.60106, 0.60118, 0.60115],
274
- [0.60451, 0.60463, 0.6046],
275
- [0.60796, 0.60809, 0.60806],
276
- [0.61141, 0.61154, 0.61151],
277
- [0.61486, 0.61499, 0.61496],
278
- [0.61833, 0.61846, 0.61843],
279
- [0.62179, 0.62192, 0.62189],
280
- [0.62527, 0.62539, 0.62536],
281
- [0.62874, 0.62887, 0.62884],
282
- [0.63222, 0.63235, 0.63231],
283
- [0.63569, 0.63583, 0.63579],
284
- [0.63918, 0.63931, 0.63928],
285
- [0.64267, 0.6428, 0.64276],
286
- [0.64615, 0.64629, 0.64625],
287
- [0.64965, 0.64978, 0.64975],
288
- [0.65314, 0.65327, 0.65324],
289
- [0.65665, 0.65678, 0.65675],
290
- [0.66015, 0.66028, 0.66025],
291
- [0.66366, 0.6638, 0.66376],
292
- [0.66717, 0.6673, 0.66727],
293
- [0.67069, 0.67082, 0.67079],
294
- [0.6742, 0.67434, 0.6743],
295
- [0.67772, 0.67786, 0.67783],
296
- [0.68124, 0.68138, 0.68134],
297
- [0.68477, 0.68491, 0.68488],
298
- [0.68831, 0.68845, 0.68841],
299
- [0.69184, 0.69198, 0.69195],
300
- [0.69538, 0.69552, 0.69548],
301
- [0.69891, 0.69906, 0.69902],
302
- [0.70246, 0.7026, 0.70257],
303
- [0.70601, 0.70616, 0.70612],
304
- [0.70956, 0.70971, 0.70967],
305
- [0.71311, 0.71326, 0.71322],
306
- [0.71668, 0.71682, 0.71679],
307
- [0.72023, 0.72038, 0.72034],
308
- [0.7238, 0.72394, 0.72391],
309
- [0.72736, 0.72751, 0.72748],
310
- [0.73093, 0.73108, 0.73105],
311
- [0.73451, 0.73466, 0.73463],
312
- [0.73808, 0.73823, 0.7382],
313
- [0.74167, 0.74182, 0.74178],
314
- [0.74525, 0.7454, 0.74536],
315
- [0.74884, 0.74899, 0.74895],
316
- [0.75243, 0.75258, 0.75254],
317
- [0.75602, 0.75618, 0.75614],
318
- [0.75961, 0.75977, 0.75973],
319
- [0.76322, 0.76337, 0.76333],
320
- [0.76682, 0.76697, 0.76694],
321
- [0.77043, 0.77058, 0.77055],
322
- [0.77403, 0.77419, 0.77415],
323
- [0.77765, 0.77781, 0.77777],
324
- [0.78126, 0.78142, 0.78138],
325
- [0.78489, 0.78504, 0.785],
326
- [0.78851, 0.78867, 0.78863],
327
- [0.79213, 0.79229, 0.79225],
328
- [0.79576, 0.79592, 0.79588],
329
- [0.79939, 0.79955, 0.79951],
330
- [0.80302, 0.80318, 0.80314],
331
- [0.80666, 0.80683, 0.80679],
332
- [0.8103, 0.81046, 0.81042],
333
- [0.81394, 0.81411, 0.81406],
334
- [0.81759, 0.81775, 0.81771],
335
- [0.82124, 0.82141, 0.82136],
336
- [0.82489, 0.82506, 0.82501],
337
- [0.82855, 0.82872, 0.82867],
338
- [0.83221, 0.83237, 0.83233],
339
- [0.83587, 0.83604, 0.83599],
340
- [0.83953, 0.8397, 0.83966],
341
- [0.8432, 0.84337, 0.84333],
342
- [0.84687, 0.84704, 0.847],
343
- [0.85054, 0.85071, 0.85067],
344
- [0.85422, 0.85439, 0.85435],
345
- [0.8579, 0.85807, 0.85803],
346
- [0.86158, 0.86176, 0.86171],
347
- [0.86527, 0.86544, 0.8654],
348
- [0.86896, 0.86913, 0.86909],
349
- [0.87265, 0.87282, 0.87278],
350
- [0.87634, 0.87652, 0.87647],
351
- [0.88004, 0.88022, 0.88017],
352
- [0.88374, 0.88392, 0.88388],
353
- [0.88744, 0.88762, 0.88758],
354
- [0.89115, 0.89133, 0.89128],
355
- [0.89486, 0.89504, 0.895],
356
- [0.89857, 0.89875, 0.89871],
357
- [0.90229, 0.90247, 0.90242],
358
- [0.90601, 0.90619, 0.90614],
359
- [0.90972, 0.90991, 0.90986],
360
- [0.91345, 0.91363, 0.91359],
361
- [0.91718, 0.91736, 0.91732],
362
- [0.9209, 0.92108, 0.92104],
363
- [0.92463, 0.92482, 0.92477],
364
- [0.92837, 0.92855, 0.92851],
365
- [0.9321, 0.93229, 0.93224],
366
- [0.93585, 0.93604, 0.93599],
367
- [0.93959, 0.93978, 0.93973],
368
- [0.94334, 0.94353, 0.94348],
369
- ]
42
+ try:
43
+ from arviz_plots import *
44
+ import arviz_plots as plots
370
45
 
46
+ _status = (
47
+ f"arviz_plots {plots.__version__} available, "
48
+ "exposing its functions as part of the `arviz` namespace"
49
+ )
50
+ _log.info(_status)
51
+ del plots
52
+ except ModuleNotFoundError as err:
53
+ raise ImportError("arviz's dependency arviz_plots is not installed", name="arviz") from err
371
54
 
372
- def _mpl_cm(name, colorlist):
373
- cmap = LinearSegmentedColormap.from_list(name, colorlist, N=256)
374
- if "cet_" + name not in mpl.colormaps():
375
- mpl.colormaps.register(cmap, name="cet_" + name)
55
+ info += _status
376
56
 
57
+ # define version last so it isn't overwritten by the respective attribute in the imported libraries
58
+ __version__ = "1.0.0rc0"
377
59
 
378
- try:
379
- import colorcet
380
- except ModuleNotFoundError:
381
- _mpl_cm("gray", _linear_grey_10_95_c0)
382
- _mpl_cm("gray_r", list(reversed(_linear_grey_10_95_c0)))
60
+ info = f"Status information for ArviZ {__version__}\n\n{info}"
383
61
 
62
+ pat = re.compile(r"arviz_(base|stats|plots)\s([0-9]+\.[0-9]+)")
63
+ matches = pat.findall(info)
64
+ if any(matches[0][1] != match[1] for match in matches[1:]):
65
+ raise ImportError(
66
+ "Versions of arviz-xyz packages don't match to the minor version. "
67
+ f"The versions found are: {matches}"
68
+ )
384
69
 
385
70
  # clean namespace
386
- del os, re, logging, version, LinearSegmentedColormap, Logger, mpl
71
+ del logging, matches, pat, re, _status
@@ -0,0 +1,182 @@
1
+ Metadata-Version: 2.4
2
+ Name: arviz
3
+ Version: 1.0.0rc0
4
+ Summary: Expose features from _ArviZverse_ refactored packages together in the ``arviz`` namespace.
5
+ Author-email: ArviZ team <arvizdevs@gmail.com>
6
+ Requires-Python: >=3.12
7
+ Description-Content-Type: text/markdown
8
+ Classifier: Development Status :: 4 - Beta
9
+ Classifier: Intended Audience :: Developers
10
+ Classifier: Intended Audience :: Science/Research
11
+ Classifier: Intended Audience :: Education
12
+ Classifier: License :: OSI Approved :: Apache Software License
13
+ Classifier: Operating System :: OS Independent
14
+ Classifier: Programming Language :: Python
15
+ Classifier: Programming Language :: Python :: 3
16
+ Classifier: Programming Language :: Python :: 3.12
17
+ Classifier: Programming Language :: Python :: 3.13
18
+ Classifier: Programming Language :: Python :: 3.14
19
+ License-File: LICENSE
20
+ Requires-Dist: arviz_base >=0.7.0
21
+ Requires-Dist: arviz_stats[xarray] >=0.7.0
22
+ Requires-Dist: arviz_plots >=0.7.0
23
+ Requires-Dist: arviz_plots[bokeh] ; extra == "bokeh"
24
+ Requires-Dist: build ; extra == "check"
25
+ Requires-Dist: pre-commit ; extra == "check"
26
+ Requires-Dist: h5netcdf ; extra == "doc"
27
+ Requires-Dist: jupyter-sphinx ; extra == "doc"
28
+ Requires-Dist: matplotlib ; extra == "doc"
29
+ Requires-Dist: myst-parser[linkify] ; extra == "doc"
30
+ Requires-Dist: myst-nb ; extra == "doc"
31
+ Requires-Dist: pydata-sphinx-theme>=0.13 ; extra == "doc"
32
+ Requires-Dist: sphinx>=5 ; extra == "doc"
33
+ Requires-Dist: sphinx-copybutton ; extra == "doc"
34
+ Requires-Dist: sphinx-design ; extra == "doc"
35
+ Requires-Dist: sphinx-notfound-page ; extra == "doc"
36
+ Requires-Dist: sphinxcontrib-youtube ; extra == "doc"
37
+ Requires-Dist: arviz_base[h5netcdf] ; extra == "h5netcdf"
38
+ Requires-Dist: arviz_plots[matplotlib] ; extra == "matplotlib"
39
+ Requires-Dist: arviz_base[netcdf4] ; extra == "netcdf4"
40
+ Requires-Dist: arviz_plots[plotly] ; extra == "plotly"
41
+ Requires-Dist: pytest ; extra == "test"
42
+ Requires-Dist: arviz_base[zarr] ; extra == "zarr"
43
+ Project-URL: documentation, https://python.arviz.org
44
+ Project-URL: funding, https://opencollective.com/arviz
45
+ Project-URL: source, https://github.com/arviz-devs/arviz
46
+ Project-URL: tracker, https://github.com/arviz-devs/arviz/issues
47
+ Provides-Extra: bokeh
48
+ Provides-Extra: check
49
+ Provides-Extra: doc
50
+ Provides-Extra: h5netcdf
51
+ Provides-Extra: matplotlib
52
+ Provides-Extra: netcdf4
53
+ Provides-Extra: plotly
54
+ Provides-Extra: test
55
+ Provides-Extra: zarr
56
+
57
+ <img src="https://raw.githubusercontent.com/arviz-devs/arviz-project/main/arviz_logos/ArviZ.png#gh-light-mode-only" width=200></img>
58
+ <img src="https://raw.githubusercontent.com/arviz-devs/arviz-project/main/arviz_logos/ArviZ_white.png#gh-dark-mode-only" width=200></img>
59
+
60
+ [![PyPI version](https://badge.fury.io/py/arviz.svg)](https://badge.fury.io/py/arviz)
61
+ [![DOI](http://joss.theoj.org/papers/10.21105/joss.01143/status.svg)](https://doi.org/10.21105/joss.01143) [![DOI](https://zenodo.org/badge/DOI/10.5281/zenodo.2540945.svg)](https://doi.org/10.5281/zenodo.2540945)
62
+ [![Powered by NumFOCUS](https://img.shields.io/badge/powered%20by-NumFOCUS-orange.svg?style=flat&colorA=E1523D&colorB=007D8A)](https://numfocus.org)
63
+
64
+ ArviZ (pronounced "AR-_vees_") is a Python package for exploratory analysis of Bayesian models. It includes functions for posterior analysis, data storage, model checking, comparison and diagnostics.
65
+
66
+ ### ArviZ in other languages
67
+ ArviZ also has a Julia wrapper available [ArviZ.jl](https://julia.arviz.org/).
68
+
69
+ ## Documentation
70
+
71
+ The ArviZ documentation can be found in the [official docs](https://python.arviz.org).
72
+ Here are some quick links for common scenarios:
73
+
74
+ * First time Bayesian modelers and ArviZ users: [EABM book](https://arviz-devs.github.io/EABM/)
75
+ * First time ArviZ users, already familiar with Bayesian modeling: [overview notebook](https://python.arviz.org/projects/plots/en/latest/tutorials/overview.html) or [example gallery](https://python.arviz.org/projects/plots/en/latest/gallery/index.html)
76
+ * ArviZ 0.x user: [migration guide](https://python.arviz.org/en/latest/user_guide/migration_guide.html)
77
+ * ArviZ-verse documentation:
78
+ - [arviz-base](https://python.arviz.org/projects/base/en/latest/)
79
+ - [arviz-stats](https://python.arviz.org/projects/stats/en/latest/)
80
+ - [arviz-plots](https://python.arviz.org/projects/plots/en/latest/)
81
+
82
+
83
+ ## Installation
84
+
85
+ ### Stable
86
+ ArviZ is available for installation from [PyPI](https://pypi.org/project/arviz/).
87
+ The latest stable version can be installed using pip:
88
+
89
+ ```
90
+ pip install "arviz[preview]"
91
+ ```
92
+
93
+ ArviZ is also available through [conda-forge](https://anaconda.org/conda-forge/arviz).
94
+
95
+ ```
96
+ conda install -c conda-forge arviz arviz-plots
97
+ ```
98
+
99
+ ### Development
100
+ The latest development version can be installed from the main branch using pip:
101
+
102
+ ```
103
+ pip install git+git://github.com/arviz-devs/arviz.git
104
+ ```
105
+
106
+ Another option is to clone the repository and install using git and setuptools:
107
+
108
+ ```
109
+ git clone https://github.com/arviz-devs/arviz.git
110
+ cd arviz
111
+ python setup.py install
112
+ ```
113
+
114
+ -------------------------------------------------------------------------------
115
+ ## [Gallery](https://python.arviz.org/en/latest/examples/index.html)
116
+
117
+ <p>
118
+ <table>
119
+ <tr>
120
+
121
+ <td>
122
+
123
+ [![plot_rank_dist example](./docs/source/_miniatures/plot_rank_dist.png)](https://arviz-plots.readthedocs.io/en/stable/api/generated/arviz_plots.plot_rank_dist.html)
124
+
125
+ </td>
126
+
127
+ <td>
128
+
129
+ [![plot_forest example with extra ESS column](./docs/source/_miniatures/plot_forest_ess.png)](https://arviz-plots.readthedocs.io/en/stable/api/generated/arviz_plots.plot_forest.html)
130
+
131
+ </td>
132
+
133
+ </tr>
134
+ </table>
135
+ <div>
136
+
137
+ <a href="https://python.arviz.org/projects/plots/en/latest/gallery/index.html">And more...</a>
138
+ </div>
139
+
140
+
141
+ ## Citation
142
+
143
+
144
+ If you use ArviZ and want to cite it please use [![DOI](http://joss.theoj.org/papers/10.21105/joss.01143/status.svg)](https://doi.org/10.21105/joss.01143)
145
+
146
+ Here is the citation in BibTeX format
147
+
148
+ ```
149
+ @article{arviz_2019,
150
+ doi = {10.21105/joss.01143},
151
+ url = {https://doi.org/10.21105/joss.01143},
152
+ year = {2019},
153
+ publisher = {The Open Journal},
154
+ volume = {4},
155
+ number = {33},
156
+ pages = {1143},
157
+ author = {Ravin Kumar and Colin Carroll and Ari Hartikainen and Osvaldo Martin},
158
+ title = {ArviZ a unified library for exploratory analysis of Bayesian models in Python},
159
+ journal = {Journal of Open Source Software}
160
+ }
161
+ ```
162
+
163
+
164
+ ## Contributions
165
+ ArviZ is a community project and welcomes contributions.
166
+ Additional information can be found in the [contributing guide](https://python.arviz.org/en/latest/contributing/index.html)
167
+
168
+
169
+ ## Code of Conduct
170
+ ArviZ wishes to maintain a positive community. Additional details
171
+ can be found in the [Code of Conduct](https://www.arviz.org/en/latest/CODE_OF_CONDUCT.html)
172
+
173
+ ## Donations
174
+ ArviZ is a non-profit project under NumFOCUS umbrella. If you want to support ArviZ financially, you can donate [here](https://numfocus.org/donate-to-arviz).
175
+
176
+ ## Sponsors and Institutional Partners
177
+ [![Aalto University](https://raw.githubusercontent.com/arviz-devs/arviz-project/refs/heads/main/cards/Aalto-black-text.png)](https://www.aalto.fi/en)
178
+ [![FCAI](https://raw.githubusercontent.com/arviz-devs/arviz-project/refs/heads/main/cards/FCAI.png)](https://fcai.fi/)
179
+ [![NumFOCUS](https://raw.githubusercontent.com/arviz-devs/arviz-project/refs/heads/main/sphinx/NumFocus.png)](https://numfocus.org)
180
+
181
+ [The ArviZ project website](https://www.arviz.org/en/latest/sponsors_partners.html) has more information about each sponsor and the support they provide.
182
+
@@ -0,0 +1,5 @@
1
+ arviz/__init__.py,sha256=0r9MN3R1VtE38yI3rgRjhl8Z7V1gPqXZ_Hk_aONLK9s,2117
2
+ arviz-1.0.0rc0.dist-info/licenses/LICENSE,sha256=tAkwu8-AdEyGxGoSvJ2gVmQdcicWw3j1ZZueVV74M-E,11357
3
+ arviz-1.0.0rc0.dist-info/WHEEL,sha256=G2gURzTEtmeR8nrdXUJfNiB3VYVxigPQ-bEQujpNiNs,82
4
+ arviz-1.0.0rc0.dist-info/METADATA,sha256=sAQznFsRxIFUdYTGd8O7sTK4wY7aZ5rZWBejY-356xY,7384
5
+ arviz-1.0.0rc0.dist-info/RECORD,,
@@ -1,5 +1,4 @@
1
1
  Wheel-Version: 1.0
2
- Generator: setuptools (80.10.2)
2
+ Generator: flit 3.12.0
3
3
  Root-Is-Purelib: true
4
4
  Tag: py3-none-any
5
-
@@ -199,4 +199,3 @@
199
199
  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
200
200
  See the License for the specific language governing permissions and
201
201
  limitations under the License.
202
-