From e529c511ba18a9ed75ed3a1d7d7bf759f95a144a Mon Sep 17 00:00:00 2001
From: Gaetan Lepage <gaetan@glepage.com>
Date: Wed, 26 Mar 2025 18:58:21 +0100
Subject: [PATCH] python312Packages.sentence-transformers: 3.4.1 -> 4.0.1

Diff: https://github.com/UKPLab/sentence-transformers/compare/refs/tags/v3.4.1...v4.0.1

Changelog: https://github.com/UKPLab/sentence-transformers/releases/tag/v4.0.1
---
 .../python-modules/sentence-transformers/default.nix   | 10 +++++++---
 1 file changed, 7 insertions(+), 3 deletions(-)

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"