From 02eb91ebd4ca8159f0d398f37b180acec8578d15 Mon Sep 17 00:00:00 2001 From: Martin Weinelt Date: Sat, 25 Jan 2025 02:12:39 +0100 Subject: [PATCH] python313Packages.accelerate: disable unsupported tests and finally migrate to pytest 8. --- .../development/python-modules/accelerate/default.nix | 11 ++++++----- 1 file changed, 6 insertions(+), 5 deletions(-) diff --git a/pkgs/development/python-modules/accelerate/default.nix b/pkgs/development/python-modules/accelerate/default.nix index ca17eda23dcb..74ab18740c0e 100644 --- a/pkgs/development/python-modules/accelerate/default.nix +++ b/pkgs/development/python-modules/accelerate/default.nix @@ -23,7 +23,7 @@ # tests evaluate, parameterized, - pytest7CheckHook, + pytestCheckHook, transformers, config, cudatoolkit, @@ -58,7 +58,7 @@ buildPythonPackage rec { nativeCheckInputs = [ evaluate parameterized - pytest7CheckHook + pytestCheckHook transformers ]; preCheck = @@ -92,10 +92,11 @@ buildPythonPackage rec { # set the environment variable, CC, which conflicts with standard environment "test_patch_environment_key_exists" ] - ++ lib.optionals (pythonAtLeast "3.12") [ - # RuntimeError: Dynamo is not supported on Python 3.12+ + ++ lib.optionals (pythonAtLeast "3.13") [ + # RuntimeError: Dynamo is not supported on Python 3.13+ + "test_can_unwrap_distributed_compiled_model_keep_torch_compile" + "test_can_unwrap_distributed_compiled_model_remove_torch_compile" "test_convert_to_fp32" - "test_dynamo_extract_model" "test_send_to_device_compiles" ] ++ lib.optionals (stdenv.hostPlatform.isLinux && stdenv.hostPlatform.isAarch64) [