From 792815c4edd9cdbe33e3f21c40e0be83b403df92 Mon Sep 17 00:00:00 2001
From: Gaetan Lepage <gaetan@glepage.com>
Date: Fri, 10 Jul 2026 21:11:58 +0000
Subject: [PATCH] python3Packages.blackjax: 1.5 -> 1.6

Diff: https://github.com/blackjax-devs/blackjax/compare/1.5...1.6

Changelog: https://github.com/blackjax-devs/blackjax/releases/tag/1.6
---
 pkgs/development/python-modules/blackjax/default.nix | 12 ++++++++----
 1 file changed, 8 insertions(+), 4 deletions(-)

diff --git a/pkgs/development/python-modules/blackjax/default.nix b/pkgs/development/python-modules/blackjax/default.nix
index df2e518d70e1..21bcd14d54fe 100644
--- a/pkgs/development/python-modules/blackjax/default.nix
+++ b/pkgs/development/python-modules/blackjax/default.nix
@@ -17,7 +17,8 @@
   typing-extensions,
 
   # optional-dependencies
-  fastprogress,
+  jax-tap,
+  tqdm,
 
   # checks
   chex,
@@ -27,14 +28,15 @@
 
 buildPythonPackage (finalAttrs: {
   pname = "blackjax";
-  version = "1.5";
+  version = "1.6";
   pyproject = true;
+  __structuredAttrs = true;
 
   src = fetchFromGitHub {
     owner = "blackjax-devs";
     repo = "blackjax";
     tag = finalAttrs.version;
-    hash = "sha256-tKJfukTqSiW2Xg3/8DakxtxlwGpJ14S/7qUE1OGM97I=";
+    hash = "sha256-qLOAmUQxr1xtlJB/TMnjFkvvHUwh0XKpPN+FVD8ju8Y=";
   };
 
   build-system = [
@@ -53,7 +55,8 @@ buildPythonPackage (finalAttrs: {
 
   optional-dependencies = {
     progress = [
-      fastprogress
+      jax-tap
+      tqdm
     ];
   };
 
@@ -77,6 +80,7 @@ buildPythonPackage (finalAttrs: {
 
     # AssertionError on numerical values
     "test_barker"
+    "test_imm_shrinkage_seed_influence_persists_diagonal"
     "test_laps"
     "test_mclmc"
     "test_mcse4"