diff options
author | Alyssa Ross <hi@alyssa.is> | 2024-06-23 00:32:22 +0200 |
---|---|---|
committer | Alyssa Ross <hi@alyssa.is> | 2024-06-23 00:32:22 +0200 |
commit | 6402b188ddd100b3cd6afe7b8a3e553365203f43 (patch) | |
tree | 676b85e4a6ffee092e413e723f7dce8ba01bb48f /nixpkgs/pkgs/development/python-modules/sentence-transformers/default.nix | |
parent | 5a1826585861b32ce2509c0643e793196d81893e (diff) | |
parent | d603719ec6e294f034936c0d0dc06f689d91b6c3 (diff) | |
download | nixlib-6402b188ddd100b3cd6afe7b8a3e553365203f43.tar nixlib-6402b188ddd100b3cd6afe7b8a3e553365203f43.tar.gz nixlib-6402b188ddd100b3cd6afe7b8a3e553365203f43.tar.bz2 nixlib-6402b188ddd100b3cd6afe7b8a3e553365203f43.tar.lz nixlib-6402b188ddd100b3cd6afe7b8a3e553365203f43.tar.xz nixlib-6402b188ddd100b3cd6afe7b8a3e553365203f43.tar.zst nixlib-6402b188ddd100b3cd6afe7b8a3e553365203f43.zip |
Diffstat (limited to 'nixpkgs/pkgs/development/python-modules/sentence-transformers/default.nix')
-rw-r--r-- | nixpkgs/pkgs/development/python-modules/sentence-transformers/default.nix | 47 |
1 files changed, 21 insertions, 26 deletions
diff --git a/nixpkgs/pkgs/development/python-modules/sentence-transformers/default.nix b/nixpkgs/pkgs/development/python-modules/sentence-transformers/default.nix index aa68ab9133e8..7b870da88879 100644 --- a/nixpkgs/pkgs/development/python-modules/sentence-transformers/default.nix +++ b/nixpkgs/pkgs/development/python-modules/sentence-transformers/default.nix @@ -1,20 +1,21 @@ -{ lib -, buildPythonPackage -, fetchFromGitHub -, huggingface-hub -, nltk -, numpy -, pytestCheckHook -, pythonOlder -, scikit-learn -, scipy -, sentencepiece -, setuptools -, tokenizers -, torch -, torchvision -, tqdm -, transformers +{ + lib, + buildPythonPackage, + fetchFromGitHub, + huggingface-hub, + nltk, + numpy, + pytestCheckHook, + pythonOlder, + scikit-learn, + scipy, + sentencepiece, + setuptools, + tokenizers, + torch, + torchvision, + tqdm, + transformers, }: buildPythonPackage rec { @@ -31,9 +32,7 @@ buildPythonPackage rec { hash = "sha256-xER+WHprW83KWJ0bom+lTn0HNU7PgGROnp/QLG1uUcw="; }; - build-system = [ - setuptools - ]; + build-system = [ setuptools ]; dependencies = [ huggingface-hub @@ -49,13 +48,9 @@ buildPythonPackage rec { transformers ]; - nativeCheckInputs = [ - pytestCheckHook - ]; + nativeCheckInputs = [ pytestCheckHook ]; - pythonImportsCheck = [ - "sentence_transformers" - ]; + pythonImportsCheck = [ "sentence_transformers" ]; disabledTests = [ # Tests require network access |