pyrox.dev
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 = "2.1.2";
15 pyproject = true;
16
17 src = fetchFromGitHub {
18 owner = "explosion";
19 repo = "spacy-curated-transformers";
20 tag = "release-v${version}";
21 hash = "sha256-Y3puV9fDN5mAugLPmXuoIbwUBpSMcmkq+oXAyYdmQew=";
22 };
23
24 build-system = [ setuptools ];
25
26 pythonRelaxDeps = [
27 "thinc"
28 ];
29
30 dependencies = [
31 curated-tokenizers
32 curated-transformers
33 spacy
34 torch
35 ];
36
37 # Unit tests are hard to use, since most tests rely on downloading
38 # models from Hugging Face Hub.
39 pythonImportsCheck = [ "spacy_curated_transformers" ];
40
41 meta = {
42 description = "spaCy entry points for Curated Transformers";
43 homepage = "https://github.com/explosion/spacy-curated-transformers";
44 changelog = "https://github.com/explosion/spacy-curated-transformers/releases/tag/v${version}";
45 license = lib.licenses.mit;
46 maintainers = with lib.maintainers; [ danieldk ];
47 };
48}