pyrox.dev
1{ lib
2, buildPythonPackage
3, fetchFromGitHub
4, bokeh
5, emcee
6, matplotlib
7, netcdf4
8, numba
9, numpy
10, pandas
11, pytest
12, setuptools
13, cloudpickle
14, pytestCheckHook
15, scipy
16, packaging
17, typing-extensions
18, pythonOlder
19, xarray
20, xarray-einstats
21, zarr
22, h5py
23, jaxlib
24, torchvision
25, jax
26 # , pymc3 (circular dependency)
27, pyro-ppl
28 #, pystan (not packaged)
29, numpyro
30}:
31
32buildPythonPackage rec {
33 pname = "arviz";
34 version = "0.14.0";
35 format = "setuptools";
36
37 disabled = pythonOlder "3.7";
38
39 src = fetchFromGitHub {
40 owner = "arviz-devs";
41 repo = pname;
42 rev = "refs/tags/${version}";
43 hash = "sha256-YLNczcgVmcctNc620Ap9yQtQTwF1LREtL57JIWS/DKQ=";
44 };
45
46 propagatedBuildInputs = [
47 matplotlib
48 netcdf4
49 numpy
50 packaging
51 pandas
52 scipy
53 setuptools
54 xarray
55 xarray-einstats
56 ];
57
58 checkInputs = [
59 bokeh
60 cloudpickle
61 emcee
62 h5py
63 jax
64 jaxlib
65 numba
66 numpyro
67 # pymc3 (circular dependency)
68 pyro-ppl
69 # pystan (not packaged)
70 pytestCheckHook
71 torchvision
72 zarr
73 ];
74
75 preCheck = ''
76 export HOME=$(mktemp -d);
77 '';
78
79 pytestFlagsArray = [
80 "arviz/tests/base_tests/"
81 ];
82
83 disabledTestPaths = [
84 # Remove tests as dependency creates a circular dependency
85 "arviz/tests/external_tests/test_data_pymc.py"
86 ];
87
88 disabledTests = [
89 # Tests require network access
90 "test_plot_separation"
91 "test_plot_trace_legend"
92 "test_cov"
93 # countourpy is not available at the moment
94 "test_plot_kde"
95 "test_plot_kde_2d"
96 "test_plot_pair"
97 ];
98
99 pythonImportsCheck = [
100 "arviz"
101 ];
102
103 meta = with lib; {
104 description = "Library for exploratory analysis of Bayesian models";
105 homepage = "https://arviz-devs.github.io/arviz/";
106 license = licenses.asl20;
107 maintainers = with maintainers; [ omnipotententity ];
108 };
109}