pyrox.dev
1{ lib
2, buildPythonPackage
3, pythonOlder
4, fetchFromGitHub
5, setuptools
6, arviz
7, formulae
8, graphviz
9, pandas
10, pymc
11, blackjax
12, numpyro
13, pytestCheckHook
14}:
15
16buildPythonPackage rec {
17 pname = "bambi";
18 version = "0.13.0";
19 pyproject = true;
20
21 disabled = pythonOlder "3.8";
22
23 src = fetchFromGitHub {
24 owner = "bambinos";
25 repo = "bambi";
26 rev = "refs/tags/${version}";
27 hash = "sha256-9+uTyV3mQlHOKAjXohwkhTzNe/+I5XR/LuH1ZYvhc8I=";
28 };
29
30 nativeBuildInputs = [
31 setuptools
32 ];
33
34 propagatedBuildInputs = [
35 arviz
36 formulae
37 graphviz
38 pandas
39 pymc
40 ];
41
42 preCheck = ''
43 export HOME=$(mktemp -d)
44 '';
45
46 nativeCheckInputs = [
47 blackjax
48 numpyro
49 pytestCheckHook
50 ];
51
52 disabledTests = [
53 # Tests require network access
54 "test_alias_equal_to_name"
55 "test_average_by"
56 "test_ax"
57 "test_basic"
58 "test_censored_response"
59 "test_custom_prior"
60 "test_data_is_copied"
61 "test_distributional_model"
62 "test_elasticity"
63 "test_extra_namespace"
64 "test_fig_kwargs"
65 "test_gamma_with_splines"
66 "test_group_effects"
67 "test_hdi_prob"
68 "test_legend"
69 "test_non_distributional_model"
70 "test_normal_with_splines"
71 "test_predict_offset"
72 "test_predict_new_groups"
73 "test_predict_new_groups_fail"
74 "test_set_alias_warnings"
75 "test_subplot_kwargs"
76 "test_transforms"
77 "test_use_hdi"
78 "test_with_groups"
79 "test_with_group_and_panel"
80 "test_with_user_values"
81 ];
82
83 pythonImportsCheck = [
84 "bambi"
85 ];
86
87 meta = with lib; {
88 homepage = "https://bambinos.github.io/bambi";
89 description = "High-level Bayesian model-building interface";
90 changelog = "https://github.com/bambinos/bambi/releases/tag/${version}";
91 license = licenses.mit;
92 maintainers = with maintainers; [ bcdarwin ];
93 };
94}