pyrox.dev
1{ lib
2, buildPythonPackage
3, fetchFromGitHub
4, huggingface-hub
5, nltk
6, numpy
7, scikit-learn
8, scipy
9, sentencepiece
10, tokenizers
11, torch
12, torchvision
13, tqdm
14, transformers
15}:
16
17buildPythonPackage rec {
18 pname = "sentence-transformers";
19 version = "2.2.2";
20 format = "setuptools";
21
22 src = fetchFromGitHub {
23 owner = "UKPLab";
24 repo = "sentence-transformers";
25 rev = "v${version}";
26 hash = "sha256-hEYpDAL0lliaS1j+c5vaZ0q1hw802jfTUurx/FvgY9w=";
27 };
28
29 propagatedBuildInputs = [
30 huggingface-hub
31 nltk
32 numpy
33 scikit-learn
34 scipy
35 sentencepiece
36 tokenizers
37 torch
38 torchvision
39 tqdm
40 transformers
41 ];
42
43 pythonImportsCheck = [ "sentence_transformers" ];
44
45 doCheck = false; # tests fail at build_ext
46
47 meta = with lib; {
48 description = "Multilingual Sentence & Image Embeddings with BERT";
49 homepage = "https://github.com/UKPLab/sentence-transformers";
50 changelog = "https://github.com/UKPLab/sentence-transformers/releases/tag/${src.rev}";
51 license = licenses.asl20;
52 maintainers = with maintainers; [ dit7ya ];
53 };
54}