Unverified Commit e6ff9e94 authored by Yt's avatar Yt Committed by GitHub
Browse files

python3Packages.sentence-transformers: 4.1.0 -> 5.0.0 (#421623)

parents eecc6a26 8fc4a3c3

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

+19 −2
Original line number Diff line number Diff line
@@ -26,14 +26,14 @@ 


buildPythonPackage rec {

  pname = "sentence-transformers";

  version = "4.1.0";

  version = "5.0.0";

  pyproject = true;



  src = fetchFromGitHub {

    owner = "UKPLab";

    repo = "sentence-transformers";

    tag = "v${version}";

    hash = "sha256-9Mg3+7Yxf195h4cUNLP/Z1PrauxanHJfS8OV2JIwRj4=";

    hash = "sha256-7HdeNyB3hMJEwHenN2hUEGG2MdQ++nF3nyAYJv7jhyA=";

  };



  build-system = [ setuptools ];
@@ -76,14 +76,26 @@ buildPythonPackage rec { 
    "test_initialization_with_embedding_dim"

    "test_initialization_with_embedding_weights"

    "test_loading_model2vec"

    "test_mine_hard_negatives_with_prompt"

    "test_model_card_base"

    "test_model_card_reuse"

    "test_nanobeir_evaluator"

    "test_paraphrase_mining"

    "test_pretrained_model"

    "test_router_as_middle_module"

    "test_router_backwards_compatibility"

    "test_router_encode"

    "test_router_load_with_config"

    "test_router_save_load"

    "test_router_save_load_with_custom_default_route"

    "test_router_save_load_with_multiple_modules_per_route"

    "test_router_save_load_without_default_route"

    "test_router_with_trainer"

    "test_router_with_trainer_without_router_mapping"

    "test_save_and_load"

    "test_simple_encode"

    "test_tokenize"

    "test_train_stsb"

    "test_trainer"

    "test_trainer_invalid_column_names"

    "test_trainer_multi_dataset_errors"
@@ -94,6 +106,11 @@ buildPythonPackage rec { 
    "tests/cross_encoder/test_cross_encoder.py"

    "tests/cross_encoder/test_train_stsb.py"

    "tests/evaluation/test_information_retrieval_evaluator.py"

    "tests/sparse_encoder/models/test_csr.py"

    "tests/sparse_encoder/models/test_sparse_static_embedding.py"

    "tests/sparse_encoder/test_opensearch_models.py"

    "tests/sparse_encoder/test_pretrained.py"

    "tests/sparse_encoder/test_sparse_encoder.py"

    "tests/test_compute_embeddings.py"

    "tests/test_model_card_data.py"

    "tests/test_multi_process.py"