gador/nixpkgs
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

python3Packages.torchtune: disable tests that fail on Darwin

Browse Source
This commit is contained in:
Sarah Clark
2025-12-24 11:16:40 -08:00
parent 900597719e
commit afdf13c128
1 changed files with 15 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pkgs/development/python-modules/torchtune/default.nix
+15
View File
@@ -114,6 +114,21 @@ buildPythonPackage rec {
++ lib.optionals (stdenv.hostPlatform.isLinux && stdenv.hostPlatform.isAarch64) [
# Fatal Python error: Segmentation fault
"test_forward_gqa"
]
++ lib.optionals stdenv.hostPlatform.isDarwin [
# tests/torchtune/training/test_distributed.py
"test_init_from_env_no_dup"
"test_init_from_env_dup"
];
disabledTestPaths = lib.optionals stdenv.hostPlatform.isDarwin [
# fail due to floating-point precision differences
"tests/torchtune/models/flux/test_flux_autoencoder.py::TestFluxAutoencoder::test_encode"
"tests/torchtune/modules/peft/test_dora.py::TestDoRALinear::test_qdora_parity[True-dtype1]"
"tests/torchtune/modules/peft/test_lora.py::TestLoRALinear::test_qlora_parity[True-dtype1]"
# hangs
"tests/torchtune/utils"
];
meta = {