diff --git a/astronomer/providers/google/cloud/hooks/dataproc.py b/astronomer/providers/google/cloud/hooks/dataproc.py index 0ffdb9bd5..1fe56f74a 100644 --- a/astronomer/providers/google/cloud/hooks/dataproc.py +++ b/astronomer/providers/google/cloud/hooks/dataproc.py @@ -3,9 +3,8 @@ from airflow.providers.google.common.consts import CLIENT_INFO from airflow.providers.google.common.hooks.base_google import GoogleBaseHook -from google.api_core import gapic_v1 +from google.api_core import gapic_v1, retry_async as retries from google.api_core.client_options import ClientOptions -from google.api_core import retry_async as retries from google.cloud.dataproc_v1 import ( ClusterControllerAsyncClient, Job,