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