python3Packages.accelerate: skip failing test on aarch64-linux (#479141)

This commit is contained in:
kirillrdy
2026-01-11 22:02:02 +00:00
committed by GitHub
@@ -31,7 +31,7 @@
writableTmpDirAsHomeHook,
}:
buildPythonPackage rec {
buildPythonPackage (finalAttrs: {
pname = "accelerate";
version = "1.12.0";
pyproject = true;
@@ -39,7 +39,7 @@ buildPythonPackage rec {
src = fetchFromGitHub {
owner = "huggingface";
repo = "accelerate";
tag = "v${version}";
tag = "v${finalAttrs.version}";
hash = "sha256-PwwaQSLOm+8Hd3trM1P+jRhYyoWM3QxOe5XT99haEmg=";
};
@@ -108,6 +108,13 @@ buildPythonPackage rec {
# TypeError: unsupported operand type(s) for /: 'NoneType' and 'int' (it seems cpuinfo doesn't work here)
"test_mpi_multicpu_config_cmd"
]
++
lib.optionals
(stdenv.hostPlatform.isLinux && stdenv.hostPlatform.isAarch64 && (pythonAtLeast "3.14"))
[
# RuntimeError: There is no current event loop in thread 'MainThread'
"test_accelerate_test"
]
++ lib.optionals (!config.cudaSupport) [
# requires ptxas from cudatoolkit, which is unfree
"test_dynamo_extract_model"
@@ -187,9 +194,9 @@ buildPythonPackage rec {
meta = {
homepage = "https://huggingface.co/docs/accelerate";
description = "Simple way to train and use PyTorch models with multi-GPU, TPU, mixed-precision";
changelog = "https://github.com/huggingface/accelerate/releases/tag/${src.tag}";
changelog = "https://github.com/huggingface/accelerate/releases/tag/${finalAttrs.src.tag}";
license = lib.licenses.asl20;
maintainers = with lib.maintainers; [ bcdarwin ];
mainProgram = "accelerate";
};
}
})