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

   1 { lib
   2 , buildPythonPackage
   3 , click
   4 , cython
   5 , fetchFromGitHub
   6 , jieba
   7 , joblib
   8 , lmdb
   9 , marisa-trie
  10 , mwparserfromhell
  11 , numpy
  12 , pythonOlder
  13 , scipy
  14 , setuptools
  15 , tqdm
  16 }:
  17
  18 buildPythonPackage rec {
  19   pname = "wikipedia2vec";
  20   version = "2.0.0";
  21   pyproject = true;
  22
  23   disabled = pythonOlder "3.8";
  24
  25   src = fetchFromGitHub {
  26     owner = "wikipedia2vec";
  27     repo = "wikipedia2vec";
  28     rev = "refs/tags/v${version}";
  29     hash = "sha256-vrBLlNm0bVIStSBWDHRCtuRpazu8JMCtBl4qJPtHGvU=";
  30   };
  31
  32   nativeBuildInputs = [
  33     cython
  34     setuptools
  35   ];
  36
  37   propagatedBuildInputs = [
  38     click
  39     cython
  40     jieba
  41     joblib
  42     lmdb
  43     marisa-trie
  44     mwparserfromhell
  45     numpy
  46     scipy
  47     tqdm
  48   ];
  49
  50   preBuild = ''
  51     bash cythonize.sh
  52   '';
  53
  54   pythonImportsCheck = [
  55     "wikipedia2vec"
  56   ];
  57
  58   meta = with lib; {
  59     description = "Tool for learning vector representations of words and entities from Wikipedia";
  60     mainProgram = "wikipedia2vec";
  61     homepage = "https://wikipedia2vec.github.io/wikipedia2vec/";
  62     changelog = "https://github.com/wikipedia2vec/wikipedia2vec/releases/tag/v${version}";
  63     license = licenses.asl20;
  64     maintainers = with maintainers; [ derdennisop ];
  65   };
  66 }