Sergey Volkov
77 lines
1.3 KiB
Nix
77 lines
1.3 KiB
Nix
{
|
|
lib,
|
|
buildPythonPackage,
|
|
fetchFromGitHub,
|
|
|
|
# build-system
|
|
cython,
|
|
setuptools,
|
|
|
|
# dependencies
|
|
numpy,
|
|
scipy,
|
|
smart-open,
|
|
|
|
# tests
|
|
mock,
|
|
pyemd,
|
|
pytestCheckHook,
|
|
testfixtures,
|
|
}:
|
|
|
|
buildPythonPackage rec {
|
|
pname = "gensim";
|
|
version = "4.4.0";
|
|
pyproject = true;
|
|
|
|
src = fetchFromGitHub {
|
|
owner = "piskvorky";
|
|
repo = "gensim";
|
|
tag = version;
|
|
hash = "sha256-TXutcU43ReBj9ss9+zBJFUxb5JqVHpl+B0c7hqcJAJY=";
|
|
};
|
|
|
|
build-system = [
|
|
cython
|
|
setuptools
|
|
];
|
|
|
|
dependencies = [
|
|
numpy
|
|
scipy
|
|
smart-open
|
|
];
|
|
|
|
nativeCheckInputs = [
|
|
mock
|
|
pyemd
|
|
pytestCheckHook
|
|
testfixtures
|
|
];
|
|
|
|
pythonImportsCheck = [ "gensim" ];
|
|
|
|
enabledTestPaths = [ "gensim/test" ];
|
|
|
|
# test_parallel is flaky under load
|
|
disabledTests = [ "test_parallel" ];
|
|
|
|
preCheck = ''
|
|
export GENSIM_DATA_DIR="$NIX_BUILD_TOP/gensim-data"
|
|
mkdir -p "$GENSIM_DATA_DIR"
|
|
export SKIP_NETWORK_TESTS=1
|
|
''
|
|
# Prevent python from importing gensim from the source files during tests
|
|
+ ''
|
|
rm gensim/__init__.py
|
|
'';
|
|
|
|
meta = {
|
|
description = "Topic-modelling library";
|
|
homepage = "https://radimrehurek.com/gensim/";
|
|
downloadPage = "https://github.com/piskvorky/gensim";
|
|
changelog = "https://github.com/RaRe-Technologies/gensim/blob/${version}/CHANGELOG.md";
|
|
license = lib.licenses.lgpl21Only;
|
|
};
|
|
}
|