diff options
| author | Pol Dellaiera <pol.dellaiera@protonmail.com> | 2024-05-11 23:05:37 +0200 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2024-05-11 23:05:37 +0200 |
| commit | 010103c926a0a9f2475f0d1ea30ba35893ca3203 (patch) | |
| tree | d8fff6cda0ae202e94b817968d4bb2251bc9a552 /pkgs | |
| parent | 7909e0f7d63ddb627867f1198ffbb09f2fe31c9f (diff) | |
| parent | 786362a76608118008455767416260af2bb1a677 (diff) | |
Merge pull request #310829 from GaetanLepage/accelerate
python311Packages.accelerate: fix/skip failing darwin tests
Diffstat (limited to 'pkgs')
| -rw-r--r-- | pkgs/development/python-modules/accelerate/default.nix | 7 | ||||
| -rw-r--r-- | pkgs/development/python-modules/gradio/default.nix | 2 |
2 files changed, 9 insertions, 0 deletions
diff --git a/pkgs/development/python-modules/accelerate/default.nix b/pkgs/development/python-modules/accelerate/default.nix index 08874ce2e0fc2..61b3c53e834cb 100644 --- a/pkgs/development/python-modules/accelerate/default.nix +++ b/pkgs/development/python-modules/accelerate/default.nix @@ -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"; diff --git a/pkgs/development/python-modules/gradio/default.nix b/pkgs/development/python-modules/gradio/default.nix index 7f3a173cf1aef..d5fc508e16229 100644 --- a/pkgs/development/python-modules/gradio/default.nix +++ b/pkgs/development/python-modules/gradio/default.nix @@ -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 = '' |