1{
2 lib,
3 buildPythonPackage,
4 fetchFromGitHub,
5 setuptools,
6 curated-tokenizers,
7 curated-transformers,
8 spacy,
9 torch,
10}:
11
12buildPythonPackage rec {
13 pname = "spacy-curated-transformers";
14 version = "0.3.0";
15 pyproject = true;
16
17 src = fetchFromGitHub {
18 owner = "explosion";
19 repo = "spacy-curated-transformers";
20 tag = "release-v${version}";
21 hash = "sha256-3LL0ofVsyacMzLJtttg0Tl9SlkPex7TwWL/HVF4WkfI=";
22 };
23
24 build-system = [ setuptools ];
25
26 dependencies = [
27 curated-tokenizers
28 curated-transformers
29 spacy
30 torch
31 ];
32
33 # Unit tests are hard to use, since most tests rely on downloading
34 # models from Hugging Face Hub.
35 pythonImportsCheck = [ "spacy_curated_transformers" ];
36
37 meta = with lib; {
38 description = "spaCy entry points for Curated Transformers";
39 homepage = "https://github.com/explosion/spacy-curated-transformers";
40 changelog = "https://github.com/explosion/spacy-curated-transformers/releases/tag/v${version}";
41 license = licenses.mit;
42 maintainers = with maintainers; [ danieldk ];
43 };
44}