diff --git a/pkgs/development/python-modules/torchtune/default.nix b/pkgs/development/python-modules/torchtune/default.nix index 7e14bd087537..1fac454501f0 100644 --- a/pkgs/development/python-modules/torchtune/default.nix +++ b/pkgs/development/python-modules/torchtune/default.nix @@ -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 = {