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