Function humanize_runtime

codeflash/code_utils/time_utils.py:7–48

(time_in_ns: int)

def humanize_runtime(time_in_ns: int) -> str:
    runtime_human: str = str(time_in_ns)
    units = "nanoseconds"
    if 1 <= time_in_ns < 2:
        units = "nanosecond"

    if time_in_ns / 1000 >= 1:
        time_micro = float(time_in_ns) / 1000
        runtime_human = humanize.precisedelta(dt.timedelta(microseconds=time_micro), minimum_unit="microseconds")

        units = re.split(r",|\s", runtime_human)[1]

        if units == "microseconds" or units == "microsecond":
            runtime_human = "%.3g" % time_micro
        elif units == "milliseconds" or units == "millisecond":
            runtime_human = "%.3g" % (time_micro / 1000)
        elif units == "seconds" or units == "second":
            runtime_human = "%.3g" % (time_micro / (1000**2))
        elif units == "minutes" or units == "minute":
            runtime_human = "%.3g" % (time_micro / (60 * 1000**2))
        else:  # hours
            runtime_human = "%.3g" % (time_micro / (3600 * 1000**2))

    runtime_human_parts = str(runtime_human).split(".")
    if len(runtime_human_parts[0]) == 1:
        if len(runtime_human_parts) == 1:
            runtime_human = f"{runtime_human_parts[0]}.00"
        elif len(runtime_human_parts[1]) >= 2:
            runtime_human = f"{runtime_human_parts[0]}.{runtime_human_parts[1][0:2]}"
        else:
            runtime_human = (
                f"{runtime_human_parts[0]}.{runtime_human_parts[1]}{'0' * (2 - len(runtime_human_parts[1]))}"
            )
    elif len(runtime_human_parts[0]) == 2:
        if len(runtime_human_parts) > 1:
            runtime_human = f"{runtime_human_parts[0]}.{runtime_human_parts[1][0]}"
        else:
            runtime_human = f"{runtime_human_parts[0]}.0"
    else:
        runtime_human = runtime_human_parts[0]

    return f"{runtime_human} {units}"