tjh.dev
1{ lib
2, buildPythonPackage
3, pythonOlder
4, fetchFromGitHub
5, pytestCheckHook
6, arviz
7, blackjax
8, formulae
9, graphviz
10, numpy
11, numpyro
12, pandas
13, pymc
14, scipy
15, setuptools
16}:
17
18buildPythonPackage rec {
19 pname = "bambi";
20 version = "0.10.0";
21 format = "pyproject";
22
23 disabled = pythonOlder "3.8";
24
25 src = fetchFromGitHub {
26 owner = "bambinos";
27 repo = pname;
28 rev = "refs/tags/${version}";
29 hash = "sha256-D04eTAlckEqgKA+59BRljlyneHYoqqZvLYmt/gBLHcU=";
30 };
31
32 nativeBuildInputs = [
33 setuptools
34 ];
35
36 propagatedBuildInputs = [
37 arviz
38 formulae
39 numpy
40 pandas
41 pymc
42 scipy
43 ];
44
45 preCheck = ''
46 export HOME=$(mktemp -d)
47 '';
48
49 nativeCheckInputs = [
50 blackjax
51 graphviz
52 numpyro
53 pytestCheckHook
54 ];
55
56 disabledTests = [
57 # Tests require network access
58 "test_custom_prior"
59 "test_data_is_copied"
60 "test_distributional_model"
61 "test_gamma_with_splines"
62 "test_non_distributional_model_with_categories"
63 "test_non_distributional_model"
64 "test_normal_with_splines"
65 "test_predict_offset"
66 # Assertion issue
67 "test_custom_likelihood_function"
68 ];
69
70 pythonImportsCheck = [
71 "bambi"
72 ];
73
74 meta = with lib; {
75 homepage = "https://bambinos.github.io/bambi";
76 description = "High-level Bayesian model-building interface";
77 license = licenses.mit;
78 maintainers = with maintainers; [ bcdarwin ];
79 };
80}