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

python3Packages.{torchao,torchtune}: disable failing tests (#494966)

Browse Source
This commit is contained in:
Gaétan Lepage
2026-03-01 11:51:43 +00:00
committed by GitHub
parent f773bf2827 16fc772d52
commit a080ab368d
2 changed files with 13 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pkgs/development/python-modules/torchao/default.nix
+6
View File
@@ -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 <class 'torch.dtype'>
"test_numerical_consistency_per_tensor"
]
++ lib.optionals (stdenv.hostPlatform.isLinux && stdenv.hostPlatform.isAarch64) [
# AssertionError: tensor(False) is not true
pkgs/development/python-modules/torchtune/default.nix
+7 -1
View File
@@ -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]"