Unverified Commit c48ca1c5 authored by Gaétan Lepage's avatar Gaétan Lepage Committed by GitHub
Browse files

python312Packages.sentence-transformers: 3.4.1 -> 4.0.1 (#393497)

parents cdd042cf e529c511

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

+7 −3
Original line number Diff line number Diff line
@@ -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"