Измерение времени выполнения через декоратор

Реализовать декоратор, который принимает на вход единицу времени (ms, s) и логирует время выполнения функции в зависимости от этой единицы.

pythonimport time

def time_duration_unit():
    pass

@time_duration_unit("ms")
def work_function():
    print("start work function work function")
    time.sleep(2)
    print("end work function. work function")

work_function()