import os

os.environ["TF_CPP_MIN_LOG_LEVEL"] = "2"

import tensorflow as tf

tf.get_logger().setLevel("ERROR")


def report_error(label, error):
    relevant = next(
        line.strip()
        for line in str(error).splitlines()
        if "Detected Infinity or NaN" in line or "Tensor had" in line
    )
    cleaned = relevant.replace("!!! ", "").replace(" !!!", "")
    if "}} " in cleaned:
        cleaned = cleaned.split("}} ", 1)[1]
    cleaned = cleaned.replace(" (# of outputs: 1)", "")
    cleaned = cleaned.split(" [Op:", 1)[0]
    print(f"{label}: {cleaned}")