tjh.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.2.4";
17 format = "setuptools";
18
19 disabled = pythonOlder "3.7";
20
21 src = fetchPypi {
22 inherit pname version;
23 hash = "sha256-hZwgk/rZ/0EAW8VcABjUKQvdYkVPdr3bzzGKroXzB7U=";
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 "transformers"
40 ];
41
42 # Test fails due to missing arguments for trfs2arrays().
43 doCheck = false;
44
45 pythonImportsCheck = [
46 "spacy_transformers"
47 ];
48
49 passthru.tests.annotation = callPackage ./annotation-test { };
50
51 meta = with lib; {
52 description = "spaCy pipelines for pretrained BERT, XLNet and GPT-2";
53 homepage = "https://github.com/explosion/spacy-transformers";
54 changelog = "https://github.com/explosion/spacy-transformers/releases/tag/v${version}";
55 license = licenses.mit;
56 maintainers = with maintainers; [ ];
57 };
58}