Unverified Commit 010103c9 authored by Pol Dellaiera's avatar Pol Dellaiera Committed by GitHub
Browse files

Merge pull request #310829 from GaetanLepage/accelerate 

python311Packages.accelerate: fix/skip failing darwin tests
parents 7909e0f7 786362a7

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

+7 −0
Original line number Diff line number Diff line
@@ -4,6 +4,7 @@ 
, fetchFromGitHub

, pythonAtLeast

, pythonOlder

, llvmPackages

, pytest7CheckHook

, setuptools

, numpy
@@ -33,6 +34,10 @@ buildPythonPackage rec { 
    hash = "sha256-E20pI5BrcTrMYrhriuOUl5/liSaQQy6eqRyCoauwb9Q=";

  };



  buildInputs = [

    llvmPackages.openmp

  ];



  build-system = [ setuptools ];



  dependencies = [
@@ -103,6 +108,8 @@ buildPythonPackage rec { 
    "accelerate"

  ];



  __darwinAllowLocalNetworking = true;



  meta = with lib; {

    homepage = "https://huggingface.co/docs/accelerate";

    description = "A simple way to train and use PyTorch models with multi-GPU, TPU, mixed-precision";

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

+2 −0
Original line number Diff line number Diff line
@@ -202,6 +202,8 @@ buildPythonPackage rec { 
    #"-W" "ignore" # uncomment for debugging help

  ];



  __darwinAllowLocalNetworking = true;



  # check the binary works outside the build env

  doInstallCheck = true;

  postInstallCheck = ''