depot/third_party/nixpkgs/pkgs/development/python-modules/pynndescent/default.nix

48 lines
908 B
Nix

{
lib,
buildPythonPackage,
fetchPypi,
importlib-metadata,
joblib,
llvmlite,
numba,
scikit-learn,
scipy,
setuptools,
pytestCheckHook,
pythonOlder,
}:
buildPythonPackage rec {
pname = "pynndescent";
version = "0.5.13";
pyproject = true;
disabled = pythonOlder "3.6";
src = fetchPypi {
inherit pname version;
hash = "sha256-10JUwO4KHu7IRZfV/on+3Pd4WT7qvjLC+XQSk0qYAPs=";
};
nativeBuildInputs = [ setuptools ];
propagatedBuildInputs = [
joblib
llvmlite
numba
scikit-learn
scipy
] ++ lib.optionals (pythonOlder "3.8") [ importlib-metadata ];
nativeCheckInputs = [ pytestCheckHook ];
pythonImportsCheck = [ "pynndescent" ];
meta = with lib; {
description = "Nearest Neighbor Descent";
homepage = "https://github.com/lmcinnes/pynndescent";
license = licenses.bsd2;
maintainers = with maintainers; [ mic92 ];
};
}