Unverified Commit eaec806a authored by Yt's avatar Yt Committed by GitHub
Browse files

python3Packages.peft: cleanup, fix on aarch64-linux; python3Packages.tinygrad:... 

python3Packages.peft: cleanup, fix on aarch64-linux; python3Packages.tinygrad: skip failing tests on aarch64-linux (#507855)
parents 35429b27 f81cee4d

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

+18 −10
Original line number Diff line number Diff line
@@ -29,7 +29,7 @@ 
  scipy,

}:



buildPythonPackage rec {

buildPythonPackage (finalAttrs: {

  pname = "peft";

  version = "0.18.1";

  pyproject = true;
@@ -37,7 +37,7 @@ buildPythonPackage rec { 
  src = fetchFromGitHub {

    owner = "huggingface";

    repo = "peft";

    tag = "v${version}";

    tag = "v${finalAttrs.version}";

    hash = "sha256-qlM8yEN/CJZbSAGNCltS4JQSzstVXRVqu47qZbLPVNc=";

  };
@@ -70,8 +70,16 @@ buildPythonPackage rec { 


  enabledTestPaths = [ "tests" ];



  disabledTests =

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

      # RuntimeError: Failed to initialize cpuinfo!

      "test_randlora_dtypes"

      "test_shira_dtypes"

      "test_vblora_dtypes"

      "test_vera_dtypes"

    ]

    ++ lib.optionals stdenv.hostPlatform.isDarwin [

      # These tests fail when MPS devices are detected

  disabledTests = lib.optional stdenv.hostPlatform.isDarwin [

      "gpu"

      "test_save_load"

      "test_resume_training_model_with_topk_weights"
@@ -113,8 +121,8 @@ buildPythonPackage rec { 
  meta = {

    homepage = "https://github.com/huggingface/peft";

    description = "State-of-the art parameter-efficient fine tuning";

    changelog = "https://github.com/huggingface/peft/releases/tag/${src.tag}";

    changelog = "https://github.com/huggingface/peft/releases/tag/${finalAttrs.src.tag}";

    license = lib.licenses.asl20;

    maintainers = with lib.maintainers; [ bcdarwin ];

  };

}
 
})

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

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

    "test_int8"

    "test_int8_to_uint16_negative"



    # RuntimeError: Failed to initialize cpuinfo!

    "test_conv2d_fused_half"

    "test_conv2d_half"

    "test_gemm_fp16"

    "test_softmax_dtype"

  ]

  ++ lib.optionals stdenv.hostPlatform.isDarwin [

    # Flaky (pass when running a smaller set of tests: tests/unit/*, but not within the full test suite)