Commit 8774cf06 authored by Gaetan Lepage's avatar Gaetan Lepage
Browse files

python311Packages.finetuning-scheduler: skip failing tests

parent 4c672880

pkgs/development/python-modules/finetuning-scheduler/default.nix

+12 −4
Original line number Diff line number Diff line
@@ -36,7 +36,15 @@ buildPythonPackage rec { 


  nativeCheckInputs = [ pytestCheckHook ];

  pytestFlagsArray = [ "tests" ];

  disabledTests = lib.optionals (stdenv.isAarch64 && stdenv.isLinux) [

  disabledTests =

    # torch._dynamo.exc.BackendCompilerFailed: backend='inductor' raised:

    # LoweringException: ImportError: cannot import name 'triton_key' from 'triton.compiler.compiler'

    lib.optionals (pythonOlder "3.12") [

      "test_fts_dynamo_enforce_p0"

      "test_fts_dynamo_resume"

      "test_fts_dynamo_intrafit"

    ]

    ++ lib.optionals (stdenv.isAarch64 && stdenv.isLinux) [

      # slightly exceeds numerical tolerance on aarch64-linux:

      "test_fts_frozen_bn_track_running_stats"

    ];