def METHOD_NAME(fn, *args, message="", log2wandb=False, **kwargs):
    start = time.time()
    value = fn(*args, **kwargs)
    duration = time.time() - start
    print(f"{message} {duration:.3f} secs")
    if log2wandb:
        wandb.log({message: duration})
    return value