NVIDIA/nvbench
1
0
Fork 0
mirror of https://github.com/NVIDIA/nvbench.git synced 2026-07-01 19:57:41 +00:00
Code Issues Packages Projects Releases Wiki Activity

colorama is not imported if --no-color is used

Browse Source
This commit is contained in:
Oleksandr Pavlyk
2026-06-29 13:03:13 -05:00
parent 4ca228ab5a
commit c2c46e90f0
2 changed files with 48 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
python/scripts/nvbench_compare.py
View File

@@ -43,7 +43,7 @@ np: Any = None
Fore: Any = None
def load_nvbench_compare_tooling() -> None:
def load_nvbench_compare_tooling(*, load_color: bool = True) -> None:
global Fore, np
if np is None:
@@ -51,7 +51,7 @@ def load_nvbench_compare_tooling() -> None:
ToolingDependency("numpy", "numpy", "bulk timing analysis"),
tool_name="nvbench-compare",
)
if Fore is None:
if load_color and Fore is None:
colorama = require_tooling_dependency(
ToolingDependency("colorama", "colorama", "colored status output"),
tool_name="nvbench-compare",
@@ -3366,7 +3366,7 @@ def main() -> int:
return 1
try:
load_nvbench_compare_tooling()
load_nvbench_compare_tooling(load_color=not args.no_color)
except MissingToolingDependencyError as exc:
print(str(exc), file=sys.stderr)
return 1
@@ -3432,7 +3432,10 @@ def main() -> int:
print(str(exc), file=sys.stderr)
return 1
warn_fore = Fore.YELLOW if args.ignore_devices else Fore.RED
if args.no_color:
warn_fore = ""
else:
warn_fore = Fore.YELLOW if args.ignore_devices else Fore.RED
msg_text = "Device sections do not match"
print(colorize(msg_text, warn_fore, Emoji.NONE, args.no_color), end="")
print(": ", end="")