Unverified Commit 89a33be7 authored by Martin Weinelt's avatar Martin Weinelt
Browse files

python3Packages.accelerate: disable failing tests on aarch64-linux 

These require cpuinfo, which is no available on aarch64-linux.
parent f0d1e982

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

+6 −0
Original line number Diff line number Diff line
@@ -107,6 +107,12 @@ buildPythonPackage (finalAttrs: { 
    "CheckpointTest"

    # TypeError: unsupported operand type(s) for /: 'NoneType' and 'int' (it seems cpuinfo doesn't work here)

    "test_mpi_multicpu_config_cmd"

    # fails cpuinfo test, because /sys/devices/system/cpu/ does not exist in the sandbox

    "test_layerwise_upcasting_inference_0"

    "test_compute_module_sizes"

    "test_compute_module_total_buffer_size"

    "test_load_checkpoint_in_model_dtype"

    "test_set_module_tensor_sets_dtype"

  ]

  ++

    lib.optionals