diff options
Diffstat (limited to 'pkgs/development/python-modules/openai-whisper/default.nix')
-rw-r--r-- | pkgs/development/python-modules/openai-whisper/default.nix | 56 |
1 files changed, 28 insertions, 28 deletions
diff --git a/pkgs/development/python-modules/openai-whisper/default.nix b/pkgs/development/python-modules/openai-whisper/default.nix index 26c2bbb0a3127..d6933e1aeea1b 100644 --- a/pkgs/development/python-modules/openai-whisper/default.nix +++ b/pkgs/development/python-modules/openai-whisper/default.nix @@ -1,27 +1,28 @@ -{ lib -, fetchFromGitHub -, buildPythonPackage -, substituteAll +{ + lib, + stdenv, + fetchFromGitHub, + buildPythonPackage, + substituteAll, -# build-system -, setuptools + # build-system + setuptools, -# runtime -, ffmpeg-headless + # runtime + ffmpeg-headless, -# propagates -, more-itertools -, numba -, numpy -, openai-triton -, scipy -, tiktoken -, torch -, tqdm -, transformers + # propagates + more-itertools, + numba, + numpy, + openai-triton, + tiktoken, + torch, + tqdm, -# tests -, pytestCheckHook + # tests + pytestCheckHook, + scipy, }: buildPythonPackage rec { @@ -43,21 +44,16 @@ buildPythonPackage rec { }) ]; - nativeBuildInputs = [ - setuptools - ]; + nativeBuildInputs = [ setuptools ]; propagatedBuildInputs = [ more-itertools numba numpy - openai-triton - scipy tiktoken torch tqdm - transformers - ]; + ] ++ lib.optionals (lib.meta.availableOn stdenv.hostPlatform openai-triton) [ openai-triton ]; preCheck = '' export HOME=$TMPDIR @@ -65,6 +61,7 @@ buildPythonPackage rec { nativeCheckInputs = [ pytestCheckHook + scipy ]; disabledTests = [ @@ -81,6 +78,9 @@ buildPythonPackage rec { mainProgram = "whisper"; homepage = "https://github.com/openai/whisper"; license = licenses.mit; - maintainers = with maintainers; [ hexa MayNiklas ]; + maintainers = with maintainers; [ + hexa + MayNiklas + ]; }; } |