Skip to content

Commit 59a5236

Browse files
Add ERROR logging prefix and sort the prefixes alphabetically (#1308)
* Add ERROR logging prefix and sort alphabetically * fix comment
1 parent 3272032 commit 59a5236

File tree

1 file changed

+8
-5
lines changed

1 file changed

+8
-5
lines changed

megatron/neox_arguments/arguments.py

+8-5
Original file line numberDiff line numberDiff line change
@@ -50,16 +50,19 @@
5050
ATTENTION_TYPE_CHOICES,
5151
)
5252

53-
### Logging colors ###
53+
### ANSI escape codes ###
54+
END = "\033[0m"
5455
GREEN = "\033[92m"
5556
RED = "\033[91m"
5657
YELLOW = "\033[93m"
57-
END = "\033[0m"
58-
SUCCESS = f"{GREEN} [SUCCESS] {END}"
59-
OKAY = f"{GREEN}[OKAY]{END}"
60-
WARNING = f"{YELLOW}[WARNING]{END}"
58+
59+
### Formatted logging prefixes ###
60+
ERROR = f"{RED}[ERROR]{END} "
6161
FAIL = f"{RED}[FAIL]{END}"
6262
INFO = "[INFO]"
63+
OKAY = f"{GREEN}[OKAY]{END}"
64+
SUCCESS = f"{GREEN} [SUCCESS] {END}"
65+
WARNING = f"{YELLOW}[WARNING]{END}"
6366

6467
# ZERO defaults by deespeed
6568
# These values should not be changed unless defaults in deepspeed are changed

0 commit comments

Comments
 (0)