From 49b2bb257f776ae7b01e3a50407e49756bebb72d Mon Sep 17 00:00:00 2001
From: Oleksandr Pavlyk <21087696+oleksandr-pavlyk@users.noreply.github.com>
Date: Mon, 29 Jun 2026 11:49:33 -0500
Subject: [PATCH] Install scripts into cuda/bench.scripts in prefix

Introduce optional dependency categories in the wheel metadata,
with cuda-bench[tools] encompasing all dependencies of nvbench
scripts.

Closes #393
---
 python/pyproject.toml | 22 ++++++++--------------
 1 file changed, 8 insertions(+), 14 deletions(-)

diff --git a/python/pyproject.toml b/python/pyproject.toml
index 6a23ff9..408a1da 100644
--- a/python/pyproject.toml
+++ b/python/pyproject.toml
@@ -42,21 +42,15 @@ test-cu13 = ["cuda-bench[cu13]", "pytest", "cupy-cuda13x", "numba"]
 
 # Generic test dependencies (defaults to CUDA 12)
 test = ["pytest", "cupy-cuda12x", "numba"]
-tools = [
-  "colorama",
-  "jsondiff",
-  "matplotlib",
-  "numpy",
-  "pandas",
-  "seaborn",
-  "tabulate",
-]
+compare = ["colorama", "jsondiff", "numpy", "tabulate"]
+plot = ["matplotlib", "numpy", "pandas", "seaborn"]
+tools = ["cuda-bench[compare,plot]"]
 
 [project.scripts]
-nvbench-compare = "scripts.nvbench_compare:main"
-nvbench-histogram = "scripts.nvbench_histogram:main"
-nvbench-json-summary = "scripts.nvbench_json_summary:main"
-nvbench-walltime = "scripts.nvbench_walltime:main"
+nvbench-compare = "cuda.bench.scripts.nvbench_compare:main"
+nvbench-histogram = "cuda.bench.scripts.nvbench_histogram:main"
+nvbench-json-summary = "cuda.bench.scripts.nvbench_json_summary:main"
+nvbench-walltime = "cuda.bench.scripts.nvbench_walltime:main"
 
 [project.urls]
 Homepage = "https://github.com/NVIDIA/nvbench"
@@ -90,4 +84,4 @@ fallback_version = "0.0.0"
 "cuda" = "cuda"
 "cuda/bench" = "cuda/bench"
 "cuda/bench/results" = "cuda/bench/results"
-"scripts" = "scripts"
+"cuda/bench/scripts" = "scripts"