From dcee4754e50d8024f6a684c9bf6e0e81972dcecd Mon Sep 17 00:00:00 2001
From: ditsuke <ditsuke@protonmail.com>
Date: Tue, 20 Aug 2024 03:05:48 +0530
Subject: [PATCH] build(nix): Add pyyaml for gguf-py

---
 .devops/nix/package-gguf-py.nix | 2 ++
 .devops/nix/scope.nix           | 2 ++
 2 files changed, 4 insertions(+)

diff --git a/.devops/nix/package-gguf-py.nix b/.devops/nix/package-gguf-py.nix
index 42d353190..083151fbc 100644
--- a/.devops/nix/package-gguf-py.nix
+++ b/.devops/nix/package-gguf-py.nix
@@ -4,6 +4,7 @@
   numpy,
   tqdm,
   sentencepiece,
+  pyyaml,
   poetry-core,
   buildPythonPackage,
   pytestCheckHook,
@@ -18,6 +19,7 @@ buildPythonPackage {
     numpy
     tqdm
     sentencepiece
+    pyyaml
   ];
   src = lib.cleanSource ../../gguf-py;
   pythonImportsCheck = [
diff --git a/.devops/nix/scope.nix b/.devops/nix/scope.nix
index 9e846c5af..478e8c422 100644
--- a/.devops/nix/scope.nix
+++ b/.devops/nix/scope.nix
@@ -11,6 +11,7 @@ let
   numpy = pythonPackages.numpy;
   tqdm = pythonPackages.tqdm;
   sentencepiece = pythonPackages.sentencepiece;
+  pyyaml = pythonPackages.pyyaml;
   poetry-core = pythonPackages.poetry-core;
   pytestCheckHook = pythonPackages.pytestCheckHook;
 in
@@ -28,6 +29,7 @@ lib.makeScope newScope (self: {
       tqdm
       sentencepiece
       poetry-core
+      pyyaml
       pytestCheckHook
       ;
   };