From 5b7aafc9501084d35745544ca88bcc7aba4f1370 Mon Sep 17 00:00:00 2001 From: pierrejeambrun Date: Sun, 23 Jul 2023 13:01:50 +0200 Subject: [PATCH] Remove unnecessary cast --- airflow/utils/providers_configuration_loader.py | 13 ++++++++----- 1 file changed, 8 insertions(+), 5 deletions(-) diff --git a/airflow/utils/providers_configuration_loader.py b/airflow/utils/providers_configuration_loader.py index df7f9d31755f1..543a25c05e6c9 100644 --- a/airflow/utils/providers_configuration_loader.py +++ b/airflow/utils/providers_configuration_loader.py @@ -17,12 +17,15 @@ from __future__ import annotations from functools import wraps -from typing import Callable, TypeVar, cast +from typing import Callable, TypeVar -T = TypeVar("T", bound=Callable) +from airflow.typing_compat import ParamSpec +PS = ParamSpec("PS") +RT = TypeVar("RT") -def providers_configuration_loaded(func: T) -> T: + +def providers_configuration_loaded(func: Callable[PS, RT]) -> Callable[PS, RT]: """ Decorator that makes sure that providers configuration is loaded before actually calling the decorated function. @@ -46,10 +49,10 @@ def providers_configuration_loaded(func: T) -> T: """ @wraps(func) - def wrapped_function(*args, **kwargs): + def wrapped_function(*args, **kwargs) -> RT: from airflow.providers_manager import ProvidersManager ProvidersManager().initialize_providers_configuration() return func(*args, **kwargs) - return cast(T, wrapped_function) + return wrapped_function