pyrox.dev
1{ lib
2, callPackage
3, fetchPypi
4, buildPythonPackage
5, pytorch
6, pythonOlder
7, spacy
8, spacy-alignments
9, srsly
10, transformers
11}:
12
13buildPythonPackage rec {
14 pname = "spacy-transformers";
15 version = "1.1.2";
16
17 disabled = pythonOlder "3.6";
18
19 src = fetchPypi {
20 inherit pname version;
21 sha256 = "b84c195dc21a28582579dea3f76c90222e29ee0d99b6adf38ade75646ed2746e";
22 };
23
24 postPatch = ''
25 sed -i 's/transformers>=3.4.0,<4.12.0/transformers/' setup.cfg
26 '';
27
28 propagatedBuildInputs = [
29 pytorch
30 spacy
31 spacy-alignments
32 srsly
33 transformers
34 ];
35
36 # Test fails due to missing arguments for trfs2arrays().
37 doCheck = false;
38
39 pythonImportsCheck = [ "spacy_transformers" ];
40
41 passthru.tests.annotation = callPackage ./annotation-test { };
42
43 meta = with lib; {
44 description = "spaCy pipelines for pretrained BERT, XLNet and GPT-2";
45 homepage = "https://github.com/explosion/spacy-transformers";
46 license = licenses.mit;
47 maintainers = with maintainers; [ ];
48 };
49}