nixpkgs/pkgs/development/python-modules/spacy/default.nix

140 lines
2.4 KiB
Nix

{
lib,
stdenv,
buildPythonPackage,
callPackage,
catalogue,
cymem,
cython_0,
fetchPypi,
hypothesis,
jinja2,
langcodes,
mock,
murmurhash,
numpy,
packaging,
preshed,
pydantic,
pytestCheckHook,
pythonOlder,
requests,
setuptools,
spacy-legacy,
spacy-loggers,
srsly,
thinc,
tqdm,
typer,
wasabi,
weasel,
writeScript,
nix,
git,
nix-update,
}:
buildPythonPackage rec {
pname = "spacy";
version = "3.8.2";
pyproject = true;
disabled = pythonOlder "3.7";
src = fetchPypi {
inherit pname version;
hash = "sha256-Szfr0lraQFmw3J4Ik+cN3l34NIUymgaO8EWA5wiSpl0=";
};
postPatch = ''
# spaCy is compatible with NumPy v1 and v2
substituteInPlace pyproject.toml setup.cfg \
--replace-fail "numpy>=2.0.0,<2.1.0" numpy
'';
build-system = [
cymem
cython_0
murmurhash
numpy
thinc
];
pythonRelaxDeps = [
"thinc"
];
dependencies = [
catalogue
cymem
jinja2
langcodes
murmurhash
numpy
packaging
preshed
pydantic
requests
setuptools
spacy-legacy
spacy-loggers
srsly
thinc
tqdm
typer
wasabi
weasel
];
nativeCheckInputs = [
pytestCheckHook
hypothesis
mock
];
# Fixes ModuleNotFoundError when running tests on Cythonized code. See #255262
preCheck = ''
cd $out
'';
pytestFlagsArray = [ "-m 'slow'" ];
disabledTests = [
# touches network
"test_download_compatibility"
"test_validate_compatibility_table"
"test_project_assets"
];
pythonImportsCheck = [ "spacy" ];
passthru = {
updateScript = writeScript "update-spacy" ''
#!${stdenv.shell}
set -eou pipefail
PATH=${
lib.makeBinPath [
nix
git
nix-update
]
}
nix-update python3Packages.spacy
# update spacy models as well
echo | nix-shell maintainers/scripts/update.nix --argstr package python3Packages.spacy-models.en_core_web_sm
'';
tests.annotation = callPackage ./annotation-test { };
};
meta = with lib; {
description = "Industrial-strength Natural Language Processing (NLP)";
mainProgram = "spacy";
homepage = "https://github.com/explosion/spaCy";
changelog = "https://github.com/explosion/spaCy/releases/tag/release-v${version}";
license = licenses.mit;
maintainers = [ ];
};
}