diff --git a/pkgs/development/python-modules/sentence-transformers/default.nix b/pkgs/development/python-modules/sentence-transformers/default.nix index 6a4a26b0ce5b..2ba0e892ac86 100644 --- a/pkgs/development/python-modules/sentence-transformers/default.nix +++ b/pkgs/development/python-modules/sentence-transformers/default.nix @@ -17,6 +17,7 @@ torch, tqdm, transformers, + typing-extensions, # tests pytestCheckHook, @@ -25,14 +26,14 @@ buildPythonPackage rec { pname = "sentence-transformers"; - version = "3.4.1"; + version = "4.0.1"; pyproject = true; src = fetchFromGitHub { owner = "UKPLab"; repo = "sentence-transformers"; tag = "v${version}"; - hash = "sha256-TNqCukHdjQYxK/UkAV/lm+TTAm5NyoZjVPUyHPyE3Ko="; + hash = "sha256-Hjf82IIkFO8e+xDK1lMp1DrkuUb6TSVQtVpmT/He/VI="; }; build-system = [ setuptools ]; @@ -45,6 +46,7 @@ buildPythonPackage rec { torch tqdm transformers + typing-extensions ]; optional-dependencies = { @@ -78,6 +80,7 @@ buildPythonPackage rec { "test_model_card_reuse" "test_nanobeir_evaluator" "test_paraphrase_mining" + "test_pretrained_model" "test_save_and_load" "test_simple_encode" "test_tokenize" @@ -88,9 +91,10 @@ buildPythonPackage rec { disabledTestPaths = [ # Tests require network access + "tests/cross_encoder/test_cross_encoder.py" + "tests/cross_encoder/test_train_stsb.py" "tests/evaluation/test_information_retrieval_evaluator.py" "tests/test_compute_embeddings.py" - "tests/test_cross_encoder.py" "tests/test_model_card_data.py" "tests/test_multi_process.py" "tests/test_pretrained_stsb.py"