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

Fix unexpected AttributeError when --no-color is used

Browse Source 
Implemented robo-review feedback suggesting to expand coverage
of lazy loading helper.
This commit is contained in:
Oleksandr Pavlyk
2026-06-29 16:34:45 -05:00
parent 92a07c53e2
commit a240460fc7
3 changed files with 90 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files

25
python/scripts/nvbench_compare.py
View File

@@ -2767,14 +2767,23 @@ def is_usable_noise(noise):
def colorize_comparison_status(status, no_color):
if status == ComparisonStatus.UNKNOWN:
return colorize(status.value, Fore.YELLOW, Emoji.YELLOW, no_color)
if status == ComparisonStatus.UNDECIDED:
return colorize(status.value, Fore.LIGHTBLACK_EX, Emoji.SHRUG, no_color)
if status == ComparisonStatus.SAME:
return colorize(status.value, Fore.BLUE, Emoji.BLUE, no_color)
if status == ComparisonStatus.FAST:
return colorize(status.value, Fore.GREEN, Emoji.GREEN, no_color)
return colorize(status.value, Fore.RED, Emoji.RED, no_color)
fore_name = "YELLOW"
emoji = Emoji.YELLOW
elif status == ComparisonStatus.UNDECIDED:
fore_name = "LIGHTBLACK_EX"
emoji = Emoji.SHRUG
elif status == ComparisonStatus.SAME:
fore_name = "BLUE"
emoji = Emoji.BLUE
elif status == ComparisonStatus.FAST:
fore_name = "GREEN"
emoji = Emoji.GREEN
else:
fore_name = "RED"
emoji = Emoji.RED
fore = "" if no_color else getattr(Fore, fore_name)
return colorize(status.value, fore, emoji, no_color)
def format_axis_values(axis_values, axes, axis_filters=None):