amoghrajesh commented on code in PR #51873:
URL: https://github.com/apache/airflow/pull/51873#discussion_r2176962041
##########
providers/amazon/src/airflow/providers/amazon/aws/hooks/base_aws.py:
##########
@@ -57,12 +57,16 @@
AirflowNotFoundException,
AirflowProviderDeprecationWarning,
)
-from airflow.hooks.base import BaseHook
from airflow.providers.amazon.aws.utils.connection_wrapper import
AwsConnectionWrapper
from airflow.providers.amazon.aws.utils.identifiers import generate_uuid
from airflow.providers.amazon.aws.utils.suppress import return_on_error
from airflow.providers.common.compat.version_compat import AIRFLOW_V_3_0_PLUS
from airflow.providers_manager import ProvidersManager
+
+try:
+ from airflow.sdk import BaseHook
+except ImportError:
+ from airflow.hooks.base import BaseHook # type: ignore
Review Comment:
Good idea, do you mind if i do that in a follow up? I am trying to avoid
working on a moving target with baseoperator effort on. Would be harder to
maintain a green CI and this PR will grow substantially.
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
