From 4c7d3841ee16363b3cf03b0745ee8f2a0007561d Mon Sep 17 00:00:00 2001
From: Gaetan Lepage <gaetan@glepage.com>
Date: Tue, 19 May 2026 12:44:57 +0000
Subject: [PATCH] python3Packages.bambi: 0.17.2 -> 0.18.0

Diff: https://github.com/bambinos/bambi/compare/0.17.2...0.18.0

Changelog: https://github.com/bambinos/bambi/releases/tag/0.18.0
---
 .../python-modules/bambi/default.nix          | 19 +++++++------------
 1 file changed, 7 insertions(+), 12 deletions(-)

diff --git a/pkgs/development/python-modules/bambi/default.nix b/pkgs/development/python-modules/bambi/default.nix
index dc271a364f8e..370c916452dd 100644
--- a/pkgs/development/python-modules/bambi/default.nix
+++ b/pkgs/development/python-modules/bambi/default.nix
@@ -15,6 +15,8 @@
   matplotlib,
   pandas,
   pymc,
+  pytensor,
+  seaborn,
   sparse,
 
   # tests
@@ -26,24 +28,16 @@
 
 buildPythonPackage (finalAttrs: {
   pname = "bambi";
-  version = "0.17.2";
+  version = "0.18.0";
   pyproject = true;
 
   src = fetchFromGitHub {
     owner = "bambinos";
     repo = "bambi";
     tag = finalAttrs.version;
-    hash = "sha256-Vjv62cYDIuTLE7MxRt4Havy7DMOiMTyIixbs4LGFGGs=";
+    hash = "sha256-vxsjPYQkqvmB5oKKl29+xq1BPEzBTozz9/W5mICWI4A=";
   };
 
-  # TypeError: NDArray.record() missing 1 required keyword-only argument: 'in_warmup'
-  postPatch = ''
-    substituteInPlace bambi/backend/pymc.py \
-      --replace-fail \
-        "strace.record(point=dict(zip(varnames, value)))" \
-        "strace.record(point=dict(zip(varnames, value)), in_warmup=False)"
-  '';
-
   build-system = [
     setuptools
     setuptools-scm
@@ -59,6 +53,8 @@ buildPythonPackage (finalAttrs: {
     matplotlib
     pandas
     pymc
+    pytensor
+    seaborn
     sparse
   ];
 
@@ -136,8 +132,7 @@ buildPythonPackage (finalAttrs: {
 
   disabledTestPaths = [
     # Tests require network access
-    "tests/test_interpret.py"
-    "tests/test_interpret_messages.py"
+    "tests/test_interpret_plots.py"
   ];
 
   pythonImportsCheck = [ "bambi" ];