python3Packages.accelerate: fix tests on darwin (#516680)

2026-05-07 14:18:57 +00:00
parent b05e90d85b f72da90b76
commit 9ab59433e2
1 changed files with 5 additions and 0 deletions
@@ -130,6 +130,11 @@ buildPythonPackage (finalAttrs: {
  ++ lib.optionals stdenv.hostPlatform.isDarwin [
    # RuntimeError: 'accelerate-launch /nix/store/a7vhm7b74a7bmxc35j26s9iy1zfaqjs...
    "test_accelerate_test"
+
+    # torch.mps does not expose a module-level set_device; keep skipped until
+    # the upstream fix lands: https://github.com/huggingface/accelerate/pull/4028
+    "test_env_var_device"
+
    "test_init_trackers"
    "test_init_trackers"
    "test_log"