pkgs/development/python-modules/spacy-transformers/default.nix

   1 { lib
   2 , callPackage
   3 , fetchPypi
   4 , buildPythonPackage
   5 , pythonRelaxDepsHook
   6 , torch
   7 , pythonOlder
   8 , spacy
   9 , spacy-alignments
  10 , srsly
  11 , transformers
  12 }:
  13
  14 buildPythonPackage rec {
  15   pname = "spacy-transformers";
  16   version = "1.3.4";
  17   format = "setuptools";
  18
  19   disabled = pythonOlder "3.7";
  20
  21   src = fetchPypi {
  22     inherit pname version;
  23     hash = "sha256-N2StqGUqOYS9mW/DAeSntNg3kii+UPdTUHDV7g1Hvus=";
  24   };
  25
  26   nativeBuildInputs = [
  27     pythonRelaxDepsHook
  28   ];
  29
  30   propagatedBuildInputs = [
  31     torch
  32     spacy
  33     spacy-alignments
  34     srsly
  35     transformers
  36   ];
  37
  38   pythonRelaxDeps = [
  39     "spacy"
  40     "transformers"
  41   ];
  42
  43   # Test fails due to missing arguments for trfs2arrays().
  44   doCheck = false;
  45
  46   pythonImportsCheck = [
  47     "spacy_transformers"
  48   ];
  49
  50   passthru.tests.annotation = callPackage ./annotation-test { };
  51
  52   meta = with lib; {
  53     description = "spaCy pipelines for pretrained BERT, XLNet and GPT-2";
  54     homepage = "https://github.com/explosion/spacy-transformers";
  55     changelog = "https://github.com/explosion/spacy-transformers/releases/tag/v${version}";
  56     license = licenses.mit;
  57     maintainers = with maintainers; [ ];
  58   };
  59 }