Декоратор, измеряющий длительность выполнения функции в заданных единицах

Необходимо реализовать декоратор, принимающий параметр — единицу измерения времени (например, "ms" или "s") — и выводящий в лог продолжительность работы функции в указанной единице.

import 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()