|
|
|
@ -2,6 +2,7 @@ |
|
|
|
|
, callPackage |
|
|
|
|
, fetchPypi |
|
|
|
|
, buildPythonPackage |
|
|
|
|
, dataclasses |
|
|
|
|
, pytorch |
|
|
|
|
, pythonOlder |
|
|
|
|
, spacy |
|
|
|
@ -13,30 +14,31 @@ |
|
|
|
|
buildPythonPackage rec { |
|
|
|
|
pname = "spacy-transformers"; |
|
|
|
|
version = "1.1.5"; |
|
|
|
|
format = "setuptools"; |
|
|
|
|
|
|
|
|
|
disabled = pythonOlder "3.6"; |
|
|
|
|
|
|
|
|
|
src = fetchPypi { |
|
|
|
|
inherit pname version; |
|
|
|
|
sha256 = "sha256-nxbmnFyHptbe5M7rQi2ECGoBpxUuutdCtY20eHsGDPI="; |
|
|
|
|
hash = "sha256-nxbmnFyHptbe5M7rQi2ECGoBpxUuutdCtY20eHsGDPI="; |
|
|
|
|
}; |
|
|
|
|
|
|
|
|
|
postPatch = '' |
|
|
|
|
sed -i 's/transformers>=3.4.0,<4.13.0/transformers/' setup.cfg |
|
|
|
|
''; |
|
|
|
|
|
|
|
|
|
propagatedBuildInputs = [ |
|
|
|
|
pytorch |
|
|
|
|
spacy |
|
|
|
|
spacy-alignments |
|
|
|
|
srsly |
|
|
|
|
transformers |
|
|
|
|
] ++ lib.optionals (pythonOlder "3.7") [ |
|
|
|
|
dataclasses |
|
|
|
|
]; |
|
|
|
|
|
|
|
|
|
# Test fails due to missing arguments for trfs2arrays(). |
|
|
|
|
doCheck = false; |
|
|
|
|
|
|
|
|
|
pythonImportsCheck = [ "spacy_transformers" ]; |
|
|
|
|
pythonImportsCheck = [ |
|
|
|
|
"spacy_transformers" |
|
|
|
|
]; |
|
|
|
|
|
|
|
|
|
passthru.tests.annotation = callPackage ./annotation-test { }; |
|
|
|
|
|
|
|
|
|