Function printf

SWF/src/utils.py:12–26

(*texts)

import numpy as np

def printf(*texts):
    if os.environ.get('DEBUG', 'True') == 'False':
        return
    if texts[0].startswith('TASK>'):
        format = "\033[1;34m{text}\033[0m"
    elif texts[0].startswith("AGENT>"):
        format = "\033[32m{text}\033[0m"
    elif texts[0].startswith('ROUND>') or texts[0].startswith('==='):
        format = "\033[1;33m{text}\033[0m"
    elif texts[0].startswith('ERROR>') or texts[0].startswith('ANSWER>') or texts[0].startswith('ENV>'):
        format = "\033[1;31m{text}\033[0m"
    else:
        format = "\033[1;37m{text}\033[0m"
    for text in texts:
        print(format.format(text=text))


def load_data(filename, file_type=None):