66 lines
1.2 KiB
Nix

{
lib,
buildPythonPackage,
fetchFromGitHub,
# build-system
setuptools,
# nativeBuildInputs
h5py,
# tests
numpy,
pytestCheckHook,
}:
buildPythonPackage rec {
pname = "annoy";
version = "1.17.3";
pyproject = true;
src = fetchFromGitHub {
owner = "spotify";
repo = "annoy";
tag = "v${version}";
hash = "sha256-oJHW4lULRun2in35pBGOKg44s5kgLH2BKiMOzVu4rf4=";
};
postPatch = ''
substituteInPlace setup.py \
--replace-fail "'nose>=1.0'" ""
'';
build-system = [ setuptools ];
nativeBuildInputs = [ h5py ];
nativeCheckInputs = [
numpy
pytestCheckHook
];
preCheck = ''
rm -rf annoy
'';
disabledTestPaths = [
# network access
"test/accuracy_test.py"
];
pythonImportsCheck = [ "annoy" ];
meta = {
description = "Approximate Nearest Neighbors in C++/Python optimized for memory usage and loading/saving to disk";
homepage = "https://github.com/spotify/annoy";
changelog = "https://github.com/spotify/annoy/releases/tag/v${version}";
license = lib.licenses.asl20;
maintainers = with lib.maintainers; [ timokau ];
badPlatforms = [
# Several tests fail with AssertionError
lib.systems.inspect.patterns.isDarwin
];
};
}