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

67 lines
1.4 KiB
Nix

{
lib,
stdenv,
buildPythonPackage,
fetchPypi,
pytestCheckHook,
ctranslate2,
ctranslate2-cpp,
sentencepiece,
stanza,
}:
let
ctranslate2OneDNN = ctranslate2.override {
ctranslate2-cpp = ctranslate2-cpp.override {
# https://github.com/OpenNMT/CTranslate2/issues/1294
withOneDNN = true;
withOpenblas = false;
};
};
in
buildPythonPackage rec {
pname = "argostranslate";
version = "1.9.6";
format = "setuptools";
src = fetchPypi {
inherit pname version;
hash = "sha256-3YzBMnqmcTIpn5UOFg3SDTFLjPSE9UDw0i8fB8LYh2s=";
};
propagatedBuildInputs = [
ctranslate2OneDNN
sentencepiece
stanza
];
postPatch = ''
ln -s */requires.txt requirements.txt
substituteInPlace requirements.txt \
--replace "==" ">="
'';
doCheck = false; # needs network access
nativeCheckInputs = [ pytestCheckHook ];
# required for import check to work
# PermissionError: [Errno 13] Permission denied: '/homeless-shelter'
env.HOME = "/tmp";
pythonImportsCheck = [
"argostranslate"
"argostranslate.translate"
];
meta = {
description = "Open-source offline translation library written in Python";
homepage = "https://www.argosopentech.com";
license = lib.licenses.mit;
maintainers = with lib.maintainers; [ misuzu ];
# Segfaults at import
broken = stdenv.hostPlatform.isDarwin && stdenv.hostPlatform.isx86_64;
};
}