diff --git a/pkgs/development/python-modules/gensim/default.nix b/pkgs/development/python-modules/gensim/default.nix index 5cf36c15dfad..3a6fc0d254a9 100644 --- a/pkgs/development/python-modules/gensim/default.nix +++ b/pkgs/development/python-modules/gensim/default.nix @@ -1,43 +1,60 @@ { lib , buildPythonPackage +, cython , fetchPypi +, mock , numpy -, six , scipy , smart-open -, scikit-learn, testfixtures, unittest2 -, isPy3k +, testfixtures +, pyemd +, pytestCheckHook +, pythonOlder }: buildPythonPackage rec { pname = "gensim"; - version = "4.1.2"; - disabled = !isPy3k; + version = "4.2.0"; + format = "setuptools"; + + disabled = pythonOlder "3.6"; src = fetchPypi { inherit pname version; - sha256 = "1932c257de4eccbb64cc40d46e8577a25f5f47b94b96019a969fb36150f11d15"; + hash = "sha256-mV69KXCjHUfBAKqsECEvR+K/EuKwZTbTiIPJUf807vE="; }; - propagatedBuildInputs = [ smart-open numpy six scipy ]; + nativeBuildInputs = [ + cython + ]; - checkInputs = [ scikit-learn testfixtures unittest2 ]; + propagatedBuildInputs = [ + smart-open + numpy + scipy + ]; - # 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 + checkInputs = [ + mock + pyemd + pytestCheckHook + ]; + + pythonImportsCheck = [ + "gensim" + ]; + + # Test setup takes several minutes doCheck = false; - meta = { + pytestFlagsArray = [ + "gensim/test" + ]; + + meta = with lib; { description = "Topic-modelling library"; homepage = "https://radimrehurek.com/gensim/"; - license = lib.licenses.lgpl21; - maintainers = with lib.maintainers; [ jyp ]; + license = licenses.lgpl21Only; + maintainers = with maintainers; [ jyp ]; }; }