Unverified Commit a080ab36 authored by Gaétan Lepage's avatar Gaétan Lepage Committed by GitHub
Browse files

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

parents f773bf28 16fc772d

pkgs/development/python-modules/torchao/default.nix

+6 −0
Original line number Diff line number Diff line
@@ -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
Original line number Diff line number Diff line
@@ -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]"