python310Packages.spacy-transformers: adjust inputs

main
Fabian Affolter 2 years ago
parent 8b176291d7
commit fc5fdba430
  1. 14
      pkgs/development/python-modules/spacy-transformers/default.nix

@ -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 { };

Loading…
Cancel
Save