Commit 9aefb679 authored by Sarah Clark's avatar Sarah Clark
Browse files

python3Packages.accelerate: disable tests that fail with an InductorError

parent f7974155

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

+7 −4
Original line number Diff line number Diff line
@@ -145,14 +145,17 @@ buildPythonPackage rec { 
      "test_state_dict_type"

      "test_with_save_limit"

      "test_with_scheduler"



      # torch._inductor.exc.InductorError: TypeError: cannot determine truth value of Relational

      "test_regional_compilation_cold_start"

      "test_regional_compilation_inference_speedup"



      # Fails in nixpkgs-review due to a port conflict with simultaneous python builds

      "test_config_compatibility"

    ]

    ++ lib.optionals (stdenv.hostPlatform.isDarwin && stdenv.hostPlatform.isx86_64) [

      # RuntimeError: torch_shm_manager: execl failed: Permission denied

      "CheckpointTest"

    ]

    ++ lib.optionals stdenv.hostPlatform.isDarwin [

      # Fails in nixpkgs-review due to a port conflict with simultaneous python builds

      "test_config_compatibility"

    ];



  disabledTestPaths = lib.optionals (!(stdenv.hostPlatform.isLinux && stdenv.hostPlatform.isx86_64)) [