pyrox.dev
1{
2 lib,
3 callPackage,
4 buildPythonPackage,
5 pythonOlder,
6 fetchFromGitHub,
7 setuptools,
8 cython,
9 pythonRelaxDepsHook,
10 spacy,
11 numpy,
12 transformers,
13 torch,
14 srsly,
15 spacy-alignments,
16 pytestCheckHook,
17}:
18
19buildPythonPackage rec {
20 pname = "spacy-transformers";
21 version = "1.3.5";
22 pyproject = true;
23
24 disabled = pythonOlder "3.7";
25
26 src = fetchFromGitHub {
27 owner = "explosion";
28 repo = "spacy-transformers";
29 rev = "refs/tags/v${version}";
30 hash = "sha256-+KCRbjY4P52SWawU1NoMoe+HOV7iujFkwqVe87fWVTE=";
31 };
32
33 build-system = [
34 setuptools
35 cython
36 ];
37
38 nativeBuildInputs = [ pythonRelaxDepsHook ];
39
40 dependencies = [
41 spacy
42 numpy
43 transformers
44 torch
45 srsly
46 spacy-alignments
47 ];
48
49 nativeCheckInputs = [ pytestCheckHook ];
50
51 pythonRelaxDeps = [ "transformers" ];
52
53 # Test fails due to missing arguments for trfs2arrays().
54 doCheck = false;
55
56 pythonImportsCheck = [ "spacy_transformers" ];
57
58 passthru.tests.annotation = callPackage ./annotation-test { };
59
60 meta = with lib; {
61 description = "spaCy pipelines for pretrained BERT, XLNet and GPT-2";
62 homepage = "https://github.com/explosion/spacy-transformers";
63 changelog = "https://github.com/explosion/spacy-transformers/releases/tag/v${version}";
64 license = licenses.mit;
65 maintainers = with maintainers; [ nickcao ];
66 };
67}