Commit fe78c282 authored by Sarah Clark's avatar Sarah Clark
Browse files

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

parent 1edca39c

pkgs/development/python-modules/sentence-transformers/default.nix

+12 −0
Original line number Diff line number Diff line
@@ -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.