pyrox.dev
1{ lib
2, buildPythonPackage
3, fetchPypi
4, numpy
5, six
6, scipy
7, smart_open
8, scikitlearn, testfixtures, unittest2
9}:
10
11buildPythonPackage rec {
12 pname = "gensim";
13 version = "3.5.0";
14 src = fetchPypi {
15 inherit pname version;
16 sha256 = "78ed9b6ac35f104542f3bee0386d71ddf9432d74c153065d2ea9f6baf10e5b49";
17 };
18
19 propagatedBuildInputs = [ smart_open numpy six scipy ];
20
21 checkInputs = [ scikitlearn testfixtures unittest2 ];
22
23 # Two tests fail.
24 #
25 # ERROR: testAddMorphemesToEmbeddings (gensim.test.test_varembed_wrapper.TestVarembed)
26 # ImportError: Could not import morfessor.
27 # This package is not in nix
28 #
29 # ERROR: testWmdistance (gensim.test.test_fasttext_wrapper.TestFastText)
30 # ImportError: Please install pyemd Python package to compute WMD.
31 # This package is not in nix
32 doCheck = false;
33
34 meta = {
35 description = "Topic-modelling library";
36 homepage = https://radimrehurek.com/gensim/;
37 license = lib.licenses.lgpl21;
38 maintainers = with lib.maintainers; [ jyp ];
39 };
40}