From 270014db5a73dd9deeab947d00b28de30bde8537 Mon Sep 17 00:00:00 2001
From: Fabian Affolter <mail@fabian-affolter.ch>
Date: Sun, 27 Oct 2024 15:51:40 +0100
Subject: [PATCH] python311Packages.bambi: 0.13.0 -> 0.14.0

Diff: https://github.com/bambinos/bambi/compare/refs/tags/0.13.0...0.14.0

Changelog: https://github.com/bambinos/bambi/releases/tag/0.14.0
---
 .../python-modules/bambi/default.nix          | 54 ++++++++++++-------
 1 file changed, 34 insertions(+), 20 deletions(-)

diff --git a/pkgs/development/python-modules/bambi/default.nix b/pkgs/development/python-modules/bambi/default.nix
index f2447b2c9d12..e85ae6b888fa 100644
--- a/pkgs/development/python-modules/bambi/default.nix
+++ b/pkgs/development/python-modules/bambi/default.nix
@@ -1,36 +1,36 @@
 {
   lib,
-  buildPythonPackage,
-  pythonOlder,
-  fetchFromGitHub,
-  setuptools,
   arviz,
+  blackjax,
+  buildPythonPackage,
+  fetchFromGitHub,
   formulae,
   graphviz,
+  numpyro,
   pandas,
   pymc,
-  blackjax,
-  numpyro,
   pytestCheckHook,
+  pythonOlder,
+  setuptools-scm,
 }:
 
 buildPythonPackage rec {
   pname = "bambi";
-  version = "0.13.0";
+  version = "0.14.0";
   pyproject = true;
 
-  disabled = pythonOlder "3.8";
+  disabled = pythonOlder "3.10";
 
   src = fetchFromGitHub {
     owner = "bambinos";
     repo = "bambi";
     rev = "refs/tags/${version}";
-    hash = "sha256-9+uTyV3mQlHOKAjXohwkhTzNe/+I5XR/LuH1ZYvhc8I=";
+    hash = "sha256-kxrNNbZfC96/XHb1I7aUHYZdFJvGR80ZI8ell/0FQXc=";
   };
 
-  nativeBuildInputs = [ setuptools ];
+  build-system = [ setuptools-scm ];
 
-  propagatedBuildInputs = [
+  dependencies = [
     arviz
     formulae
     graphviz
@@ -38,9 +38,10 @@ buildPythonPackage rec {
     pymc
   ];
 
-  preCheck = ''
-    export HOME=$(mktemp -d)
-  '';
+  # bayeux-ml is not available in nixpkgs
+  # optional-dependencies = {
+  #   jax = [ bayeux-ml ];
+  # };
 
   nativeCheckInputs = [
     blackjax
@@ -48,6 +49,10 @@ buildPythonPackage rec {
     pytestCheckHook
   ];
 
+  preCheck = ''
+    export HOME=$(mktemp -d)
+  '';
+
   disabledTests = [
     # Tests require network access
     "test_alias_equal_to_name"
@@ -65,29 +70,38 @@ buildPythonPackage rec {
     "test_group_effects"
     "test_hdi_prob"
     "test_legend"
+    "test_model_with_group_specific_effects"
+    "test_model_with_intercept"
+    "test_model_without_intercept"
     "test_non_distributional_model"
     "test_normal_with_splines"
-    "test_predict_offset"
-    "test_predict_new_groups"
     "test_predict_new_groups_fail"
+    "test_predict_new_groups"
+    "test_predict_offset"
     "test_set_alias_warnings"
     "test_subplot_kwargs"
     "test_transforms"
     "test_use_hdi"
-    "test_with_groups"
     "test_with_group_and_panel"
+    "test_with_groups"
     "test_with_user_values"
   ];
 
+  disabledTestPaths = [
+    # bayeux-ml is not available
+    "tests/test_alternative_samplers.py"
+    # Tests require network access
+    "tests/test_interpret.py"
+    "tests/test_interpret_messages.py"
+  ];
+
   pythonImportsCheck = [ "bambi" ];
 
   meta = with lib; {
-    homepage = "https://bambinos.github.io/bambi";
     description = "High-level Bayesian model-building interface";
+    homepage = "https://bambinos.github.io/bambi";
     changelog = "https://github.com/bambinos/bambi/releases/tag/${version}";
     license = licenses.mit;
     maintainers = with maintainers; [ bcdarwin ];
-    # https://github.com/NixOS/nixpkgs/issues/310940
-    broken = true;
   };
 }