def pretty_time(delta: dt.timedelta) -> str:
"""
Args:
delta: a delta between two dt.datetime objects
Returns:
a pretty string of the time delta
"""
delta = str(delta)
if delta.startswith("0:00:00."):
delta_us = int(delta.split(".")[1])
delta_ms = int(delta_us / 1000)
if delta_ms >= 500:
return f"\033[91m{delta_ms}ms\033[0m"
else:
return f"{delta_ms}ms"
elif delta.startswith("0:00:01."):
delta_us = int(delta.split(".")[1])
delta_ms = 1000 + int(delta_us / 1000)
return f"\033[91m{delta_ms}ms\033[0m"
else:
return f"\033[91m{delta_ms}\033[0m"
