diff options
author | Daniƫl de Kok <me@danieldk.eu> | 2021-05-16 10:01:08 +0200 |
---|---|---|
committer | Jonathan Ringer <jonringer@users.noreply.github.com> | 2021-05-17 23:29:28 -0700 |
commit | e15fd57929a77f8d69ecee4f2f2718cb000a0f33 (patch) | |
tree | 965385d53d8f6c6ac1d920a64d77d5b4e1d3f99c /pkgs/development/python-modules/spacy-transformers | |
parent | a3405dfa2624ba5800ffb914922649b590cc2ba5 (diff) |
python3Packages.spacy-transformers: init at 1.0.2
This package provides spaCy model pipelines that wrap Hugging Face's transformers package, so that they can be used in spaCy.
Diffstat (limited to 'pkgs/development/python-modules/spacy-transformers')
-rw-r--r-- | pkgs/development/python-modules/spacy-transformers/default.nix | 39 |
1 files changed, 39 insertions, 0 deletions
diff --git a/pkgs/development/python-modules/spacy-transformers/default.nix b/pkgs/development/python-modules/spacy-transformers/default.nix new file mode 100644 index 0000000000000..0e8669d98fff7 --- /dev/null +++ b/pkgs/development/python-modules/spacy-transformers/default.nix @@ -0,0 +1,39 @@ +{ lib +, fetchPypi +, buildPythonPackage +, pytorch +, spacy +, spacy-alignments +, srsly +, transformers +}: + +buildPythonPackage rec { + pname = "spacy-transformers"; + version = "1.0.2"; + + src = fetchPypi { + inherit pname version; + hash = "sha256-AYshH2trMTgeSkAPRb6wRWpm4gA5FaKV2NJd+PhzAy4="; + }; + + propagatedBuildInputs = [ + pytorch + spacy + spacy-alignments + srsly + transformers + ]; + + # Test fails due to missing arguments for trfs2arrays(). + doCheck = false; + + pythonImportsCheck = [ "spacy_transformers" ]; + + meta = with lib; { + description = "spaCy pipelines for pretrained BERT, XLNet and GPT-2"; + homepage = "https://github.com/explosion/spacy-transformers"; + license = licenses.mit; + maintainers = with maintainers; [ danieldk ]; + }; +} |