pkgs/development/python-modules/stanza/default.nix

   1 {
   2   lib,
   3   buildPythonPackage,
   4   emoji,
   5   fetchFromGitHub,
   6   networkx,
   7   numpy,
   8   peft,
   9   protobuf,
  10   pythonOlder,
  11   requests,
  12   six,
  13   toml,
  14   torch,
  15   tqdm,
  16   transformers,
  17 }:
  18
  19 buildPythonPackage rec {
  20   pname = "stanza";
  21   version = "1.9.2";
  22   format = "setuptools";
  23
  24   disabled = pythonOlder "3.8";
  25
  26   src = fetchFromGitHub {
  27     owner = "stanfordnlp";
  28     repo = pname;
  29     rev = "refs/tags/v${version}";
  30     hash = "sha256-hrRn6ITsN7kFL1T6VjSXPDytANEeJYKwMaSdCG+YJyM=";
  31   };
  32
  33   propagatedBuildInputs = [
  34     emoji
  35     networkx
  36     numpy
  37     peft
  38     protobuf
  39     requests
  40     six
  41     toml
  42     torch
  43     tqdm
  44     transformers
  45   ];
  46
  47   # Tests require network access
  48   doCheck = false;
  49
  50   pythonImportsCheck = [ "stanza" ];
  51
  52   meta = with lib; {
  53     description = "Official Stanford NLP Python Library for Many Human Languages";
  54     homepage = "https://github.com/stanfordnlp/stanza/";
  55     changelog = "https://github.com/stanfordnlp/stanza/releases/tag/v${version}";
  56     license = licenses.asl20;
  57     maintainers = with maintainers; [ riotbib ];
  58   };
  59 }