training-transformers-together/dashboard-embedded

1from functools import wraps2from time import time3 4 5def simple_time_tracker(log_fun):6    def _simple_time_tracker(fn):7        @wraps(fn)8        def wrapped_fn(*args, **kwargs):9            start_time = time()10 11            try:12                result = fn(*args, **kwargs)13            finally:14                elapsed_time = time() - start_time15 16                # log the result17                log_fun(18                    {19                        "function_name": fn.__name__,20                        "total_time": elapsed_time,21                    }22                )23 24            return result25 26        return wrapped_fn27 28    return _simple_time_tracker29 30 31def _log(message):32    print("[SimpleTimeTracker] {function_name} {total_time:.3f}".format(**message))33