build(nix): Add pyyaml for gguf-py

This commit is contained in:
ditsuke 2024-08-20 03:05:48 +05:30
parent 06a8547a20
commit dcee4754e5
No known key found for this signature in database
GPG key ID: 71B6C31C8A5A9D21
2 changed files with 4 additions and 0 deletions

View file

@ -4,6 +4,7 @@
numpy, numpy,
tqdm, tqdm,
sentencepiece, sentencepiece,
pyyaml,
poetry-core, poetry-core,
buildPythonPackage, buildPythonPackage,
pytestCheckHook, pytestCheckHook,
@ -18,6 +19,7 @@ buildPythonPackage {
numpy numpy
tqdm tqdm
sentencepiece sentencepiece
pyyaml
]; ];
src = lib.cleanSource ../../gguf-py; src = lib.cleanSource ../../gguf-py;
pythonImportsCheck = [ pythonImportsCheck = [

View file

@ -11,6 +11,7 @@ let
numpy = pythonPackages.numpy; numpy = pythonPackages.numpy;
tqdm = pythonPackages.tqdm; tqdm = pythonPackages.tqdm;
sentencepiece = pythonPackages.sentencepiece; sentencepiece = pythonPackages.sentencepiece;
pyyaml = pythonPackages.pyyaml;
poetry-core = pythonPackages.poetry-core; poetry-core = pythonPackages.poetry-core;
pytestCheckHook = pythonPackages.pytestCheckHook; pytestCheckHook = pythonPackages.pytestCheckHook;
in in
@ -28,6 +29,7 @@ lib.makeScope newScope (self: {
tqdm tqdm
sentencepiece sentencepiece
poetry-core poetry-core
pyyaml
pytestCheckHook pytestCheckHook
; ;
}; };