diff --git a/pkgs/development/python-modules/torchao/default.nix b/pkgs/development/python-modules/torchao/default.nix index 152c2c90b1e4..224b3ba8c95b 100644 --- a/pkgs/development/python-modules/torchao/default.nix +++ b/pkgs/development/python-modules/torchao/default.nix @@ -153,6 +153,12 @@ buildPythonPackage (finalAttrs: { "test_qdq_per_channel" "test_reentrant" "test_static_linear" + + # AttributeError: 'list' object has no attribute 'keys' + "test_tied_weights_quantization" + + # execnet.gateway_base.DumpError: can't serialize + "test_numerical_consistency_per_tensor" ] ++ lib.optionals (stdenv.hostPlatform.isLinux && stdenv.hostPlatform.isAarch64) [ # AssertionError: tensor(False) is not true diff --git a/pkgs/development/python-modules/torchtune/default.nix b/pkgs/development/python-modules/torchtune/default.nix index df6795d344f9..9072cdb7bb92 100644 --- a/pkgs/development/python-modules/torchtune/default.nix +++ b/pkgs/development/python-modules/torchtune/default.nix @@ -128,7 +128,13 @@ buildPythonPackage (finalAttrs: { "test_init_from_env_dup" ]; - disabledTestPaths = lib.optionals stdenv.hostPlatform.isDarwin [ + disabledTestPaths = [ + # TypeError: HfHubHTTPError.__init__() missing 1 required keyword-only argument: 'response' + "tests/torchtune/_cli/test_download.py::TestTuneDownloadCommand::test_download_calls_snapshot" + "tests/torchtune/_cli/test_download.py::TestTuneDownloadCommand::test_gated_repo_error_no_token" + "tests/torchtune/_cli/test_download.py::TestTuneDownloadCommand::test_gated_repo_error_with_token" + ] + ++ 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]"