Skip to content

Log

Logging utilities for console output with labels and colors.

This module provides helper functions for printing status messages with labels and consistent formatting, supporting both local and distributed execution.

generate_random_string()

Generate a random string of length 10

Source code in mlir_rl_artifact/utils/log.py 
19
20
21
def generate_random_string():
    """Generate a random string of length 10"""
    return ''.join(random.choices(string.ascii_letters + string.digits, k=10))

print_info(*args, add_label=True, **kwargs)

Prints an information message

Source code in mlir_rl_artifact/utils/log.py 
29
30
31
32
33
34
def print_info(*args, add_label: bool = True, **kwargs):
    """Prints an information message"""
    message = ' '.join(map(str, args))
    label = f'{time_log()} - [INFO]    ' if add_label else ''
    for line in message.split('\n'):
        print(f"\033[94m{label}{line}\033[0m", **kwargs)

print_success(*args, add_label=True, **kwargs)

Prints a success message

Source code in mlir_rl_artifact/utils/log.py 
37
38
39
40
41
42
def print_success(*args, add_label: bool = True, **kwargs):
    """Prints a success message"""
    message = ' '.join(map(str, args))
    label = f'{time_log()} - [SUCCESS]    ' if add_label else ''
    for line in message.split('\n'):
        print(f"\033[92m{label}{line}\033[0m", **kwargs)

print_alert(*args, add_label=True, **kwargs)

Prints an alert message

Source code in mlir_rl_artifact/utils/log.py 
45
46
47
48
49
50
def print_alert(*args, add_label: bool = True, **kwargs):
    """Prints an alert message"""
    message = ' '.join(map(str, args))
    label = f'{time_log()} - [ALERT]    ' if add_label else ''
    for line in message.split('\n'):
        print(f"\033[93m{label}{line}\033[0m", file=sys.stderr, **kwargs)

print_error(*args, add_label=True, with_barrier=True, **kwargs)

Prints an error message

Source code in mlir_rl_artifact/utils/log.py 
53
54
55
56
57
58
59
60
def print_error(*args, add_label: bool = True, with_barrier: bool = True, **kwargs):
    """Prints an error message"""
    message = ' '.join(map(str, args))
    if with_barrier:
        message = '\n----------------------------------------\n' + message + '\n----------------------------------------\n'
    label = f'{time_log()} - [ERROR]    ' if add_label else ''
    for line in message.split('\n'):
        print(f"\033[91m{label}{line}\033[0m", file=sys.stderr, **kwargs)