Commit 36acad3d authored by Ben Darwin's avatar Ben Darwin
Browse files

python310Packages.bambi: enable more tests

parent 20f49fa1

pkgs/development/python-modules/bambi/default.nix

+8 −4
Original line number Diff line number Diff line
@@ -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" ];