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) [