From cc1ac71df5506424d9a6f27969515bc930e08d74 Mon Sep 17 00:00:00 2001 From: Kirill Radzikhovskyy Date: Sat, 26 Oct 2024 06:50:47 +1100 Subject: python312Packages.llama-cpp-python: add passthru.test to build with CUDA support --- pkgs/development/python-modules/llama-cpp-python/default.nix | 2 ++ 1 file changed, 2 insertions(+) (limited to 'pkgs/development/python-modules/llama-cpp-python') diff --git a/pkgs/development/python-modules/llama-cpp-python/default.nix b/pkgs/development/python-modules/llama-cpp-python/default.nix index 9a2f9458a875..30eaf7ee6b77 100644 --- 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"; -- cgit v1.2.3