ctr-ll4 / src /utils /decorators.py
sanjin7's picture
Upload src/ with huggingface_hub
cea4a4b
raw
history blame contribute delete
388 Bytes
import time
from datetime import timedelta
import pandas as pd
from loguru import logger
def print_execution_time(func):
def wrapper_time(*args, **kwargs):
start = time.perf_counter()
a = func(*args, **kwargs)
end = time.perf_counter()
logger.debug(f"{func.__name__}: {str(timedelta(seconds=end - start))}")
return a
return wrapper_time