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

   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
  16 buildPythonPackage 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 }