about summary refs log tree commit diff
path: root/pkgs/development/python-modules/fairseq/default.nix
diff options
context:
space:
mode:
Diffstat (limited to 'pkgs/development/python-modules/fairseq/default.nix')
-rw-r--r--pkgs/development/python-modules/fairseq/default.nix65
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 ];
   };
 }