Gaetan Lepage
2025-03-27 02:05:04 +01:00
parent 372754b5f0
commit e529c511ba
@@ -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"