From 486f894778e397ba891a22b3a7325ebdbd0898bb Mon Sep 17 00:00:00 2001
From: Gaetan Lepage <gaetan@glepage.com>
Date: Sat, 27 Dec 2025 17:21:15 +0000
Subject: [PATCH] python3Packages.dm-sonnet: only collect tests under sonnet/

---
 pkgs/development/python-modules/dm-sonnet/default.nix | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/pkgs/development/python-modules/dm-sonnet/default.nix b/pkgs/development/python-modules/dm-sonnet/default.nix
index 7b201bffdb70..a113283d5591 100644
--- a/pkgs/development/python-modules/dm-sonnet/default.nix
+++ b/pkgs/development/python-modules/dm-sonnet/default.nix
@@ -67,6 +67,12 @@ buildPythonPackage rec {
     export TF_USE_LEGACY_KERAS=True
   '';
 
+  enabledTestPaths = [
+    # Prevent collecting docs/ext/link_tf_api_test.py which fails with:
+    # ModuleNotFoundError: No module named 'docs.ext'
+    "sonnet"
+  ];
+
   disabledTests = [
     # AssertionError: 2 != 0 : 2 doctests failed
     "test_doctest_sonnet.functional"