python3Packages.dm-sonnet: only collect tests under sonnet/

2025-12-27 17:21:15 +00:00
parent f8a9902584
commit 486f894778
1 changed files with 6 additions and 0 deletions
@@ -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"