python314Packages.sentence-transformers: disable failing tests under python 3.14

2026-01-05 16:18:55 -08:00
parent 1edca39caf
commit fe78c282a3
1 changed files with 12 additions and 0 deletions
@@ -3,6 +3,7 @@
  stdenv,
  buildPythonPackage,
  fetchFromGitHub,
+  pythonAtLeast,

  # build-system
  setuptools,
@@ -128,6 +129,17 @@ buildPythonPackage rec {
    # NameError: name 'ParallelismConfig' is not defined
    "test_hf_argument_parser"
    "test_hf_argument_parser_incorrect_string_arguments"
+
+  ]
+  ++ lib.optionals (pythonAtLeast "3.14") [
+    # TypeError: Pickler._batch_setitems() takes 2 positional arguments but 3 were given
+    # https://github.com/huggingface/sentence-transformers/issues/3606
+    "test_group_by_label_batch_sampler_label_a"
+    "test_group_by_label_batch_sampler_label_b"
+    "test_group_by_label_batch_sampler_uneven_dataset"
+    "test_proportional_no_duplicates"
+    "test_round_robin_batch_sampler"
+    "test_round_robin_batch_sampler_vallue_error"
  ]
  ++ lib.optionals (!stdenv.hostPlatform.isAarch64 && stdenv.hostPlatform.isDarwin) [
    # These sparse tests also time out, on x86_64-darwin.