diff options
Diffstat (limited to 'pkgs/development/python-modules/fairseq/default.nix')
-rw-r--r-- | pkgs/development/python-modules/fairseq/default.nix | 65 |
1 files changed, 31 insertions, 34 deletions
diff --git a/pkgs/development/python-modules/fairseq/default.nix b/pkgs/development/python-modules/fairseq/default.nix index e784f8ec0b0d..5cea23e2bda6 100644 --- a/pkgs/development/python-modules/fairseq/default.nix +++ b/pkgs/development/python-modules/fairseq/default.nix @@ -1,32 +1,32 @@ -{ lib -, buildPythonPackage -, pythonOlder -, fetchFromGitHub -, fetchpatch - -# Native build inputs -, cython -, pythonRelaxDepsHook -, which - -# Propagated build inputs -, cffi -, hydra-core -, omegaconf -, sacrebleu -, numpy -, regex -, torch -, tqdm -, bitarray -, torchaudio -, scikit-learn -, packaging - -# Check inputs -, expecttest -, hypothesis -, pytestCheckHook +{ + lib, + buildPythonPackage, + pythonOlder, + fetchFromGitHub, + fetchpatch, + + # Native build inputs + cython, + which, + + # Propagated build inputs + cffi, + hydra-core, + omegaconf, + sacrebleu, + numpy, + regex, + torch, + tqdm, + bitarray, + torchaudio, + scikit-learn, + packaging, + + # Check inputs + expecttest, + hypothesis, + pytestCheckHook, }: buildPythonPackage rec { @@ -52,7 +52,6 @@ buildPythonPackage rec { nativeBuildInputs = [ cython - pythonRelaxDepsHook which ]; @@ -89,9 +88,7 @@ buildPythonPackage rec { cd tests ''; - pytestFlagsArray = [ - "--import-mode append" - ]; + pytestFlagsArray = [ "--import-mode append" ]; disabledTests = [ # this test requires xformers @@ -120,7 +117,7 @@ buildPythonPackage rec { homepage = "https://github.com/pytorch/fairseq"; license = licenses.mit; platforms = platforms.linux; - hydraPlatforms = []; + hydraPlatforms = [ ]; maintainers = with maintainers; [ happysalada ]; }; } |