python312Packages.llama-cpp-python: add passthru.test to build with CUDA

support
NixOS · Oct 25, 2024 · 4173566 · 4173566
1 parent 3b20a6b
commit 4173566
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/pkgs/development/python-modules/llama-cpp-python/default.nix b/pkgs/development/python-modules/llama-cpp-python/default.nix
@@ -10,6 +10,7 @@
   pytestCheckHook,
   pythonOlder,
   scikit-build-core,
+  llama-cpp-python,
 
   config,
   cudaSupport ? config.cudaSupport,
@@ -86,6 +87,7 @@ buildPythonPackage rec {
   pythonImportsCheck = [ "llama_cpp" ];
 
   passthru.updateScript = gitUpdater { rev-prefix = "v"; };
+  passthru.tests.llama-cpp-python = llama-cpp-python.override { cudaSupport = true; };
 
   meta = {
     description = "Python bindings for llama.cpp";