Commit f3d401a1 authored by Sarah Clark's avatar Sarah Clark
Browse files

python3Packages.torchao: disable failing tests

parent 8582b135

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