diff options
Diffstat (limited to 'verif/checker/tosa_result_checker.py')
-rw-r--r-- | verif/checker/tosa_result_checker.py | 11 |
1 files changed, 9 insertions, 2 deletions
diff --git a/verif/checker/tosa_result_checker.py b/verif/checker/tosa_result_checker.py index 3a15de9..66864c2 100644 --- a/verif/checker/tosa_result_checker.py +++ b/verif/checker/tosa_result_checker.py @@ -11,7 +11,7 @@ from pathlib import Path import numpy as np ################################## -no_color_printing = False +color_printing = True @unique @@ -25,9 +25,16 @@ class LogColors(Enum): BOLD_WHITE = "\u001b[1m" +def set_print_in_color(enabled): + """Set color printing to enabled or disabled.""" + global color_printing + color_printing = enabled + + def print_color(color, msg): """Print color status messages if enabled.""" - if no_color_printing: + global color_printing + if not color_printing: print(msg) else: print("{}{}{}".format(color.value, msg, LogColors.NONE.value)) |