From c4916f86c9316c9db3f890ea8a327bab25dc28e5 Mon Sep 17 00:00:00 2001
From: Kirill Radzikhovskyy <kirillrdy@gmail.com>
Date: Tue, 30 Dec 2025 07:01:40 +1100
Subject: [PATCH] python3Packages.tensorflow-probability: unbreak build with
 bazel 7

---
 .../python-modules/tensorflow-probability/default.nix       | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/pkgs/development/python-modules/tensorflow-probability/default.nix b/pkgs/development/python-modules/tensorflow-probability/default.nix
index 9e23f1c0e5a2..755a7bee60c7 100644
--- a/pkgs/development/python-modules/tensorflow-probability/default.nix
+++ b/pkgs/development/python-modules/tensorflow-probability/default.nix
@@ -78,10 +78,12 @@ let
     bazel = bazel;
 
     bazelTargets = [ ":pip_pkg" ];
+    bazelFlags = [ "--noenable_bzlmod" ];
+    removeRulesCC = false;
     LIBTOOL = lib.optionalString stdenv.hostPlatform.isDarwin "${cctools}/bin/libtool";
 
     fetchAttrs = {
-      sha256 = "sha256-TbWcWYidyXuAMgBnO2/k0NKCzc4wThf2uUeC3QxdBJY=";
+      sha256 = "sha256-7sPdIHWNFn13eaUanFgN988hFAwGnlU6cxmHOJUDpiQ=";
     };
 
     buildAttrs = {
@@ -146,7 +148,5 @@ buildPythonPackage {
     changelog = "https://github.com/tensorflow/probability/releases/tag/v${version}";
     license = lib.licenses.asl20;
     maintainers = with lib.maintainers; [ GaetanLepage ];
-    # Needs update for Bazel 7.
-    broken = true;
   };
 }