Декоратор, изменяющий возвращаемое значение + type hints

# вернет значение 10, а не 8
@convert_return(lambda x: x+2)
def my_value(n):
    return 8

# вернет список от 0 до 9, а не генератор
@convert_return(list)
def my_range(n):
    for i in range(10):
        yield i
 
# вернет словарь вида {'0': 0, '1': 100, '2': 200, ..,}
@convert_return(dict)
def my_dict(n):
    for i in range(10):
        yield str(i), i*100

def convert_return(conversion_fun):
    def real_decorator(real_function):
        def wrapper(*args, **kwargs):
            result = real_function(*args, **kwargs)
            return conversion_fun(result)
 
        return wrapper
 
    return real_decorator

F = TypeVar('F')
 
def convert_return(conversion_fun: F) -> Callable[[F], F]:
    def real_decorator(real_function: F) -> F:
        def wrapper(*args, **kwargs):
            result = real_function(*args, **kwargs)
            return conversion_fun(result)
 
        return wrapper
 
    return real_decorator

T1 = TypeVar('T1')
T2 = TypeVar('T2')
 
OriginalFun = Callable[..., T1]
ResultFun = Callable[..., T2]
ConversionFun = Callable[[T1], T2]
DecoratorFun = Callable[[OriginalFun], ResultFun]
 
def convert_return(converter_fun: ConversionFun) -> DecoratorFun:
    def decorator(original_fun: OriginalFun) -> ResultFun:
        def result_fun(*args: Any, **kwargs: Any) -> T2:
            result = original_fun(*args, **kwargs)
            return converter_fun(result)
 
        return result_fun
 
    return decorator