python3Packages.accelerate: fix tests on darwin

2026-05-04 19:58:43 -04:00
parent 54ff9097ff
commit f72da90b76
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"