summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorBen Darwin <bcdarwin@gmail.com>2023-02-06 21:56:05 -0500
committerBen Darwin <bcdarwin@gmail.com>2023-02-06 21:59:16 -0500
commit36acad3daf6fa6cc32fed4641eaf7e36984f055b (patch)
tree46cd5b9814554501410bb4053301540fd0b14ec4
parentpython310Packages.blackjax: init at 0.9.6 (diff)
downloadnixpkgs-36acad3daf6fa6cc32fed4641eaf7e36984f055b.tar.gz
python310Packages.bambi: enable more tests
-rw-r--r--pkgs/development/python-modules/bambi/default.nix12
1 files changed, 8 insertions, 4 deletions
diff --git a/pkgs/development/python-modules/bambi/default.nix b/pkgs/development/python-modules/bambi/default.nix
index c15c3117b69b..2999eadd678f 100644
--- a/pkgs/development/python-modules/bambi/default.nix
+++ b/pkgs/development/python-modules/bambi/default.nix
@@ -4,9 +4,11 @@
, fetchFromGitHub
, pytestCheckHook
, arviz
+, blackjax
, formulae
, graphviz
, numpy
+, numpyro
, pandas
, pymc
, scipy
@@ -35,14 +37,16 @@ buildPythonPackage rec {
preCheck = ''export HOME=$(mktemp -d)'';
- nativeCheckInputs = [ graphviz pytestCheckHook ];
+ nativeCheckInputs = [
+ blackjax
+ graphviz
+ numpyro
+ pytestCheckHook
+ ];
disabledTests = [
# attempt to fetch data:
"test_data_is_copied"
"test_predict_offset"
- # require blackjax (not in Nixpkgs), numpyro, and jax:
- "test_logistic_regression_categoric_alternative_samplers"
- "test_regression_alternative_samplers"
];
pythonImportsCheck = [ "bambi" ];