diff options
Diffstat (limited to 'pkgs/development/python-modules/openai-whisper/default.nix')
-rw-r--r-- | pkgs/development/python-modules/openai-whisper/default.nix | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/pkgs/development/python-modules/openai-whisper/default.nix b/pkgs/development/python-modules/openai-whisper/default.nix index 0038098e586ab..88f0ffab5dc7e 100644 --- a/pkgs/development/python-modules/openai-whisper/default.nix +++ b/pkgs/development/python-modules/openai-whisper/default.nix @@ -1,4 +1,5 @@ { lib +, stdenv , fetchFromGitHub , buildPythonPackage , substituteAll @@ -44,17 +45,17 @@ buildPythonPackage rec { nativeBuildInputs = [ setuptools - scipy ]; propagatedBuildInputs = [ more-itertools numba numpy - openai-triton tiktoken torch tqdm + ] ++ lib.optionals (lib.meta.availableOn stdenv.hostPlatform openai-triton) [ + openai-triton ]; preCheck = '' @@ -63,6 +64,7 @@ buildPythonPackage rec { nativeCheckInputs = [ pytestCheckHook + scipy ]; disabledTests = [ |