nixpkgs-suyu/pkgs/development/python-modules/gensim/default.nix
2017-08-25 19:36:14 +02:00

43 lines
1.1 KiB
Nix

{ lib
, buildPythonPackage
, fetchPypi
, numpy
, six
, scipy
, smart_open
, scikitlearn
, testfixtures
, unittest2
}:
buildPythonPackage rec {
pname = "gensim";
name = "${pname}-${version}";
version = "2.3.0";
src = fetchPypi {
inherit pname version;
sha256 = "7d0dccc7d2c576e270037949874800b7cfbc86ef081ff981483f612cd18e223f";
};
propagatedBuildInputs = [ smart_open numpy six scipy
# scikitlearn testfixtures unittest2 # for tests
];
doCheck = false;
# Two tests fail.
# ERROR: testAddMorphemesToEmbeddings (gensim.test.test_varembed_wrapper.TestVarembed)
# ImportError: Could not import morfessor.
# This package is not in nix
# ERROR: testWmdistance (gensim.test.test_fasttext_wrapper.TestFastText)
# ImportError: Please install pyemd Python package to compute WMD.
# This package is not in nix
meta = {
description = "Topic-modelling library";
homepage = https://radimrehurek.com/gensim/;
license = lib.licenses.lgpl21;
maintainers = with lib.maintainers; [ jpbernardy ];
};
}