From c7383a8e78e163acb7b0d170b2ec195f60cc05dc Mon Sep 17 00:00:00 2001
From: Ben Darwin <bcdarwin@gmail.com>
Date: Fri, 28 Apr 2023 11:40:53 -0400
Subject: [PATCH] python310Packages.accelerate: unbreak

Disable known failing test on torch>=2.0
---
 pkgs/development/python-modules/accelerate/default.nix | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/pkgs/development/python-modules/accelerate/default.nix b/pkgs/development/python-modules/accelerate/default.nix
index be841e9bb4d5..3edf016c9fa1 100644
--- a/pkgs/development/python-modules/accelerate/default.nix
+++ b/pkgs/development/python-modules/accelerate/default.nix
@@ -53,6 +53,9 @@ buildPythonPackage rec {
     # try to download data:
     "FeatureExamplesTests"
     "test_infer_auto_device_map_on_t0pp"
+    # known failure with Torch>2.0; see https://github.com/huggingface/accelerate/pull/1339:
+    # (remove for next release)
+    "test_gradient_sync_cpu_multi"
   ] ++ lib.optionals (stdenv.isLinux && stdenv.isAarch64) [
     # usual aarch64-linux RuntimeError: DataLoader worker (pid(s) <...>) exited unexpectedly
     "CheckpointTest"