apache-airflow-providers-google 14.0.0__py3-none-any.whl → 14.1.0rc1__py3-none-any.whl

airflow/providers/google/cloud/operators/vertex_ai/hyperparameter_tuning_job.py

@@ -23,6 +23,10 @@ from __future__ import annotations
 from collections.abc import Sequence
 from typing import TYPE_CHECKING, Any
 
+from google.api_core.exceptions import NotFound
+from google.api_core.gapic_v1.method import DEFAULT, _MethodDefault
+from google.cloud.aiplatform_v1 import types
+
 from airflow.configuration import conf
 from airflow.exceptions import AirflowException
 from airflow.providers.google.cloud.hooks.vertex_ai.hyperparameter_tuning_job import (
@@ -34,15 +38,13 @@ from airflow.providers.google.cloud.links.vertex_ai import (
 )
 from airflow.providers.google.cloud.operators.cloud_base import GoogleCloudBaseOperator
 from airflow.providers.google.cloud.triggers.vertex_ai import CreateHyperparameterTuningJobTrigger
-from google.api_core.exceptions import NotFound
-from google.api_core.gapic_v1.method import DEFAULT, _MethodDefault
-from google.cloud.aiplatform_v1 import types
 
 if TYPE_CHECKING:
-    from airflow.utils.context import Context
     from google.api_core.retry import Retry
     from google.cloud.aiplatform import HyperparameterTuningJob, gapic, hyperparameter_tuning
 
+    from airflow.utils.context import Context
+
 
 class CreateHyperparameterTuningJobOperator(GoogleCloudBaseOperator):
     """

airflow/providers/google/cloud/operators/vertex_ai/model_service.py

@@ -22,6 +22,10 @@ from __future__ import annotations
 from collections.abc import Sequence
 from typing import TYPE_CHECKING
 
+from google.api_core.exceptions import NotFound
+from google.api_core.gapic_v1.method import DEFAULT, _MethodDefault
+from google.cloud.aiplatform_v1.types import Model, model_service
+
 from airflow.providers.google.cloud.hooks.vertex_ai.model_service import ModelServiceHook
 from airflow.providers.google.cloud.links.vertex_ai import (
     VertexAIModelExportLink,
@@ -29,14 +33,12 @@ from airflow.providers.google.cloud.links.vertex_ai import (
     VertexAIModelListLink,
 )
 from airflow.providers.google.cloud.operators.cloud_base import GoogleCloudBaseOperator
-from google.api_core.exceptions import NotFound
-from google.api_core.gapic_v1.method import DEFAULT, _MethodDefault
-from google.cloud.aiplatform_v1.types import Model, model_service
 
 if TYPE_CHECKING:
-    from airflow.utils.context import Context
     from google.api_core.retry import Retry
 
+    from airflow.utils.context import Context
+
 
 class DeleteModelOperator(GoogleCloudBaseOperator):
     """

airflow/providers/google/cloud/operators/vertex_ai/pipeline_job.py

@@ -23,6 +23,10 @@ from collections.abc import Sequence
 from functools import cached_property
 from typing import TYPE_CHECKING, Any
 
+from google.api_core.exceptions import NotFound
+from google.api_core.gapic_v1.method import DEFAULT, _MethodDefault
+from google.cloud.aiplatform_v1 import types
+
 from airflow.configuration import conf
 from airflow.exceptions import AirflowException
 from airflow.providers.google.cloud.hooks.vertex_ai.pipeline_job import PipelineJobHook
@@ -32,16 +36,14 @@ from airflow.providers.google.cloud.links.vertex_ai import (
 )
 from airflow.providers.google.cloud.operators.cloud_base import GoogleCloudBaseOperator
 from airflow.providers.google.cloud.triggers.vertex_ai import RunPipelineJobTrigger
-from google.api_core.exceptions import NotFound
-from google.api_core.gapic_v1.method import DEFAULT, _MethodDefault
-from google.cloud.aiplatform_v1 import types
 
 if TYPE_CHECKING:
-    from airflow.utils.context import Context
     from google.api_core.retry import Retry
     from google.cloud.aiplatform import PipelineJob
     from google.cloud.aiplatform.metadata import experiment_resources
 
+    from airflow.utils.context import Context
+
 
 class RunPipelineJobOperator(GoogleCloudBaseOperator):
     """

airflow/providers/google/cloud/operators/video_intelligence.py

@@ -22,16 +22,18 @@ from __future__ import annotations
 from collections.abc import Sequence
 from typing import TYPE_CHECKING
 
-from airflow.providers.google.cloud.hooks.video_intelligence import CloudVideoIntelligenceHook
-from airflow.providers.google.cloud.operators.cloud_base import GoogleCloudBaseOperator
 from google.api_core.gapic_v1.method import DEFAULT, _MethodDefault
 from google.cloud.videointelligence_v1 import Feature, VideoContext
 from google.protobuf.json_format import MessageToDict
 
+from airflow.providers.google.cloud.hooks.video_intelligence import CloudVideoIntelligenceHook
+from airflow.providers.google.cloud.operators.cloud_base import GoogleCloudBaseOperator
+
 if TYPE_CHECKING:
-    from airflow.utils.context import Context
     from google.api_core.retry import Retry
 
+    from airflow.utils.context import Context
+
 
 class CloudVideoIntelligenceDetectVideoLabelsOperator(GoogleCloudBaseOperator):
     """

airflow/providers/google/cloud/operators/vision.py

@@ -23,9 +23,6 @@ from collections.abc import Sequence
 from copy import deepcopy
 from typing import TYPE_CHECKING, Any
 
-from airflow.providers.google.cloud.hooks.vision import CloudVisionHook
-from airflow.providers.google.cloud.operators.cloud_base import GoogleCloudBaseOperator
-from airflow.providers.google.common.hooks.base_google import PROVIDE_PROJECT_ID
 from google.api_core.exceptions import AlreadyExists
 from google.api_core.gapic_v1.method import DEFAULT, _MethodDefault
 from google.cloud.vision_v1 import (
@@ -36,11 +33,16 @@ from google.cloud.vision_v1 import (
     ReferenceImage,
 )
 
+from airflow.providers.google.cloud.hooks.vision import CloudVisionHook
+from airflow.providers.google.cloud.operators.cloud_base import GoogleCloudBaseOperator
+from airflow.providers.google.common.hooks.base_google import PROVIDE_PROJECT_ID
+
 if TYPE_CHECKING:
-    from airflow.utils.context import Context
     from google.api_core.retry import Retry
     from google.protobuf.field_mask_pb2 import FieldMask
 
+    from airflow.utils.context import Context
+
 
 MetaData = Sequence[tuple[str, str]]
 

airflow/providers/google/cloud/operators/workflows.py

@@ -23,6 +23,11 @@ import uuid
 from collections.abc import Sequence
 from typing import TYPE_CHECKING
 
+from google.api_core.exceptions import AlreadyExists
+from google.api_core.gapic_v1.method import DEFAULT, _MethodDefault
+from google.cloud.workflows.executions_v1beta import Execution
+from google.cloud.workflows_v1beta import Workflow
+
 from airflow.providers.google.cloud.hooks.workflows import WorkflowsHook
 from airflow.providers.google.cloud.links.workflows import (
     WorkflowsExecutionLink,
@@ -31,16 +36,13 @@ from airflow.providers.google.cloud.links.workflows import (
 )
 from airflow.providers.google.cloud.operators.cloud_base import GoogleCloudBaseOperator
 from airflow.providers.google.common.hooks.base_google import PROVIDE_PROJECT_ID
-from google.api_core.exceptions import AlreadyExists
-from google.api_core.gapic_v1.method import DEFAULT, _MethodDefault
-from google.cloud.workflows.executions_v1beta import Execution
-from google.cloud.workflows_v1beta import Workflow
 
 if TYPE_CHECKING:
-    from airflow.utils.context import Context
     from google.api_core.retry import Retry
     from google.protobuf.field_mask_pb2 import FieldMask
 
+    from airflow.utils.context import Context
+
 from airflow.utils.hashlib_wrapper import md5
 
 

airflow/providers/google/cloud/secrets/secret_manager.py

@@ -21,6 +21,8 @@ from __future__ import annotations
 import logging
 from collections.abc import Sequence
 
+from google.auth.exceptions import DefaultCredentialsError
+
 from airflow.exceptions import AirflowException
 from airflow.providers.google.cloud._internal_client.secret_manager_client import _SecretManagerClient
 from airflow.providers.google.cloud.utils.credentials_provider import (
@@ -30,7 +32,6 @@ from airflow.providers.google.cloud.utils.credentials_provider import (
 from airflow.providers.google.common.hooks.base_google import PROVIDE_PROJECT_ID
 from airflow.secrets import BaseSecretsBackend
 from airflow.utils.log.logging_mixin import LoggingMixin
-from google.auth.exceptions import DefaultCredentialsError
 
 log = logging.getLogger(__name__)
 

airflow/providers/google/cloud/sensors/bigquery_dts.py

@@ -22,17 +22,19 @@ from __future__ import annotations
 from collections.abc import Sequence
 from typing import TYPE_CHECKING
 
+from google.api_core.gapic_v1.method import DEFAULT, _MethodDefault
+from google.cloud.bigquery_datatransfer_v1 import TransferState
+
 from airflow.exceptions import AirflowException
 from airflow.providers.google.cloud.hooks.bigquery_dts import BiqQueryDataTransferServiceHook
 from airflow.providers.google.common.hooks.base_google import PROVIDE_PROJECT_ID
 from airflow.sensors.base import BaseSensorOperator
-from google.api_core.gapic_v1.method import DEFAULT, _MethodDefault
-from google.cloud.bigquery_datatransfer_v1 import TransferState
 
 if TYPE_CHECKING:
-    from airflow.utils.context import Context
     from google.api_core.retry import Retry
 
+    from airflow.utils.context import Context
+
 
 class BigQueryDataTransferServiceTransferRunSensor(BaseSensorOperator):
     """

airflow/providers/google/cloud/sensors/bigtable.py

@@ -23,13 +23,14 @@ from collections.abc import Sequence
 from typing import TYPE_CHECKING
 
 import google.api_core.exceptions
+from google.cloud.bigtable import enums
+from google.cloud.bigtable.table import ClusterState
+
 from airflow.providers.google.cloud.hooks.bigtable import BigtableHook
 from airflow.providers.google.cloud.links.bigtable import BigtableTablesLink
 from airflow.providers.google.cloud.operators.bigtable import BigtableValidationMixin
 from airflow.providers.google.common.hooks.base_google import PROVIDE_PROJECT_ID
 from airflow.sensors.base import BaseSensorOperator
-from google.cloud.bigtable import enums
-from google.cloud.bigtable.table import ClusterState
 
 if TYPE_CHECKING:
     from airflow.utils.context import Context

airflow/providers/google/cloud/sensors/cloud_composer.py

@@ -26,6 +26,7 @@ from functools import cached_property
 from typing import TYPE_CHECKING
 
 from dateutil import parser
+from google.cloud.orchestration.airflow.service_v1.types import Environment, ExecuteAirflowCommandResponse
 
 from airflow.configuration import conf
 from airflow.exceptions import AirflowException
@@ -34,7 +35,6 @@ from airflow.providers.google.cloud.triggers.cloud_composer import CloudComposer
 from airflow.providers.google.common.consts import GOOGLE_DEFAULT_DEFERRABLE_METHOD_NAME
 from airflow.sensors.base import BaseSensorOperator
 from airflow.utils.state import TaskInstanceState
-from google.cloud.orchestration.airflow.service_v1.types import Environment, ExecuteAirflowCommandResponse
 
 if TYPE_CHECKING:
     from airflow.utils.context import Context

airflow/providers/google/cloud/sensors/dataplex.py

@@ -23,9 +23,14 @@ from collections.abc import Sequence
 from typing import TYPE_CHECKING
 
 if TYPE_CHECKING:
-    from airflow.utils.context import Context
     from google.api_core.retry import Retry
 
+    from airflow.utils.context import Context
+
+from google.api_core.exceptions import GoogleAPICallError
+from google.api_core.gapic_v1.method import DEFAULT, _MethodDefault
+from google.cloud.dataplex_v1.types import DataScanJob
+
 from airflow.exceptions import AirflowException
 from airflow.providers.google.cloud.hooks.dataplex import (
     AirflowDataQualityScanException,
@@ -33,9 +38,6 @@ from airflow.providers.google.cloud.hooks.dataplex import (
     DataplexHook,
 )
 from airflow.sensors.base import BaseSensorOperator
-from google.api_core.exceptions import GoogleAPICallError
-from google.api_core.gapic_v1.method import DEFAULT, _MethodDefault
-from google.cloud.dataplex_v1.types import DataScanJob
 
 
 class TaskState:

airflow/providers/google/cloud/sensors/dataproc.py

@@ -23,12 +23,13 @@ import time
 from collections.abc import Sequence
 from typing import TYPE_CHECKING
 
+from google.api_core.exceptions import ServerError
+from google.cloud.dataproc_v1.types import Batch, JobStatus
+
 from airflow.exceptions import AirflowException
 from airflow.providers.google.cloud.hooks.dataproc import DataprocHook
 from airflow.providers.google.common.hooks.base_google import PROVIDE_PROJECT_ID
 from airflow.sensors.base import BaseSensorOperator
-from google.api_core.exceptions import ServerError
-from google.cloud.dataproc_v1.types import Batch, JobStatus
 
 if TYPE_CHECKING:
     from airflow.utils.context import Context

airflow/providers/google/cloud/sensors/dataproc_metastore.py

@@ -26,9 +26,10 @@ from airflow.providers.google.cloud.hooks.gcs import parse_json_from_gcs
 from airflow.sensors.base import BaseSensorOperator
 
 if TYPE_CHECKING:
-    from airflow.utils.context import Context
     from google.api_core.operation import Operation
 
+    from airflow.utils.context import Context
+
 
 class MetastoreHivePartitionSensor(BaseSensorOperator):
     """

airflow/providers/google/cloud/sensors/gcs.py

@@ -25,6 +25,8 @@ from collections.abc import Sequence
 from datetime import datetime, timedelta
 from typing import TYPE_CHECKING, Any, Callable
 
+from google.cloud.storage.retry import DEFAULT_RETRY
+
 from airflow.configuration import conf
 from airflow.exceptions import AirflowException
 from airflow.providers.google.cloud.hooks.gcs import GCSHook
@@ -35,12 +37,12 @@ from airflow.providers.google.cloud.triggers.gcs import (
     GCSUploadSessionTrigger,
 )
 from airflow.sensors.base import BaseSensorOperator, poke_mode_only
-from google.cloud.storage.retry import DEFAULT_RETRY
 
 if TYPE_CHECKING:
-    from airflow.utils.context import Context
     from google.api_core.retry import Retry
 
+    from airflow.utils.context import Context
+
 
 class GCSObjectExistenceSensor(BaseSensorOperator):
     """

airflow/providers/google/cloud/sensors/pubsub.py

@@ -23,13 +23,14 @@ from collections.abc import Sequence
 from datetime import timedelta
 from typing import TYPE_CHECKING, Any, Callable
 
+from google.cloud import pubsub_v1
+from google.cloud.pubsub_v1.types import ReceivedMessage
+
 from airflow.configuration import conf
 from airflow.exceptions import AirflowException
 from airflow.providers.google.cloud.hooks.pubsub import PubSubHook
 from airflow.providers.google.cloud.triggers.pubsub import PubsubPullTrigger
 from airflow.sensors.base import BaseSensorOperator
-from google.cloud import pubsub_v1
-from google.cloud.pubsub_v1.types import ReceivedMessage
 
 if TYPE_CHECKING:
     from airflow.utils.context import Context

airflow/providers/google/cloud/sensors/workflows.py

@@ -19,17 +19,19 @@ from __future__ import annotations
 from collections.abc import Sequence
 from typing import TYPE_CHECKING
 
+from google.api_core.gapic_v1.method import DEFAULT, _MethodDefault
+from google.cloud.workflows.executions_v1beta import Execution
+
 from airflow.exceptions import AirflowException
 from airflow.providers.google.cloud.hooks.workflows import WorkflowsHook
 from airflow.providers.google.common.hooks.base_google import PROVIDE_PROJECT_ID
 from airflow.sensors.base import BaseSensorOperator
-from google.api_core.gapic_v1.method import DEFAULT, _MethodDefault
-from google.cloud.workflows.executions_v1beta import Execution
 
 if TYPE_CHECKING:
-    from airflow.utils.context import Context
     from google.api_core.retry import Retry
 
+    from airflow.utils.context import Context
+
 
 class WorkflowExecutionSensor(BaseSensorOperator):
     """

airflow/providers/google/cloud/transfers/bigquery_to_gcs.py

@@ -22,6 +22,9 @@ from __future__ import annotations
 from collections.abc import Sequence
 from typing import TYPE_CHECKING, Any
 
+from google.api_core.exceptions import Conflict
+from google.cloud.bigquery import DEFAULT_RETRY, UnknownJob
+
 from airflow.configuration import conf
 from airflow.exceptions import AirflowException
 from airflow.models import BaseOperator
@@ -30,13 +33,12 @@ from airflow.providers.google.cloud.links.bigquery import BigQueryTableLink
 from airflow.providers.google.cloud.triggers.bigquery import BigQueryInsertJobTrigger
 from airflow.providers.google.common.hooks.base_google import PROVIDE_PROJECT_ID
 from airflow.utils.helpers import merge_dicts
-from google.api_core.exceptions import Conflict
-from google.cloud.bigquery import DEFAULT_RETRY, UnknownJob
 
 if TYPE_CHECKING:
-    from airflow.utils.context import Context
     from google.api_core.retry import Retry
 
+    from airflow.utils.context import Context
+
 
 class BigQueryToGCSOperator(BaseOperator):
     """

airflow/providers/google/cloud/transfers/gcs_to_bigquery.py

@@ -23,15 +23,6 @@ import json
 from collections.abc import Sequence
 from typing import TYPE_CHECKING, Any
 
-from airflow.configuration import conf
-from airflow.exceptions import AirflowException
-from airflow.models import BaseOperator
-from airflow.providers.google.cloud.hooks.bigquery import BigQueryHook, BigQueryJob
-from airflow.providers.google.cloud.hooks.gcs import GCSHook
-from airflow.providers.google.cloud.links.bigquery import BigQueryTableLink
-from airflow.providers.google.cloud.triggers.bigquery import BigQueryInsertJobTrigger
-from airflow.providers.google.common.hooks.base_google import PROVIDE_PROJECT_ID
-from airflow.utils.helpers import merge_dicts
 from google.api_core.exceptions import BadRequest, Conflict
 from google.cloud.bigquery import (
     DEFAULT_RETRY,
@@ -45,10 +36,21 @@ from google.cloud.bigquery import (
 )
 from google.cloud.bigquery.table import EncryptionConfiguration, Table, TableReference
 
+from airflow.configuration import conf
+from airflow.exceptions import AirflowException
+from airflow.models import BaseOperator
+from airflow.providers.google.cloud.hooks.bigquery import BigQueryHook, BigQueryJob
+from airflow.providers.google.cloud.hooks.gcs import GCSHook
+from airflow.providers.google.cloud.links.bigquery import BigQueryTableLink
+from airflow.providers.google.cloud.triggers.bigquery import BigQueryInsertJobTrigger
+from airflow.providers.google.common.hooks.base_google import PROVIDE_PROJECT_ID
+from airflow.utils.helpers import merge_dicts
+
 if TYPE_CHECKING:
-    from airflow.utils.context import Context
     from google.api_core.retry import Retry
 
+    from airflow.utils.context import Context
+
 ALLOWED_FORMATS = [
     "CSV",
     "NEWLINE_DELIMITED_JSON",

airflow/providers/google/cloud/triggers/bigquery_dts.py

@@ -21,9 +21,10 @@ import asyncio
 from collections.abc import AsyncIterator, Sequence
 from typing import Any
 
+from google.cloud.bigquery_datatransfer_v1 import TransferRun, TransferState
+
 from airflow.providers.google.cloud.hooks.bigquery_dts import AsyncBiqQueryDataTransferServiceHook
 from airflow.triggers.base import BaseTrigger, TriggerEvent
-from google.cloud.bigquery_datatransfer_v1 import TransferRun, TransferState
 
 
 class BigQueryDataTransferRunTrigger(BaseTrigger):

airflow/providers/google/cloud/triggers/cloud_batch.py

@@ -20,9 +20,10 @@ import asyncio
 from collections.abc import AsyncIterator, Sequence
 from typing import Any
 
+from google.cloud.batch_v1 import Job, JobStatus
+
 from airflow.providers.google.cloud.hooks.cloud_batch import CloudBatchAsyncHook
 from airflow.triggers.base import BaseTrigger, TriggerEvent
-from google.cloud.batch_v1 import Job, JobStatus
 
 DEFAULT_BATCH_LOCATION = "us-central1"
 

airflow/providers/google/cloud/triggers/cloud_build.py

@@ -20,9 +20,10 @@ import asyncio
 from collections.abc import AsyncIterator, Sequence
 from typing import Any
 
+from google.cloud.devtools.cloudbuild_v1.types import Build
+
 from airflow.providers.google.cloud.hooks.cloud_build import CloudBuildAsyncHook
 from airflow.triggers.base import BaseTrigger, TriggerEvent
-from google.cloud.devtools.cloudbuild_v1.types import Build
 
 
 class CloudBuildCreateBuildTrigger(BaseTrigger):

airflow/providers/google/cloud/triggers/cloud_composer.py

@@ -25,11 +25,11 @@ from datetime import datetime
 from typing import Any
 
 from dateutil import parser
+from google.cloud.orchestration.airflow.service_v1.types import ExecuteAirflowCommandResponse
 
 from airflow.exceptions import AirflowException
 from airflow.providers.google.cloud.hooks.cloud_composer import CloudComposerAsyncHook
 from airflow.triggers.base import BaseTrigger, TriggerEvent
-from google.cloud.orchestration.airflow.service_v1.types import ExecuteAirflowCommandResponse
 
 
 class CloudComposerExecutionTrigger(BaseTrigger):

airflow/providers/google/cloud/triggers/cloud_storage_transfer_service.py

@@ -21,6 +21,9 @@ import asyncio
 from collections.abc import AsyncIterator, Iterable, Sequence
 from typing import Any
 
+from google.api_core.exceptions import GoogleAPIError
+from google.cloud.storage_transfer_v1.types import TransferOperation
+
 from airflow.exceptions import AirflowException
 from airflow.providers.google.cloud.hooks.cloud_storage_transfer_service import (
     CloudDataTransferServiceAsyncHook,
@@ -28,8 +31,6 @@ from airflow.providers.google.cloud.hooks.cloud_storage_transfer_service import
 )
 from airflow.providers.google.common.hooks.base_google import PROVIDE_PROJECT_ID
 from airflow.triggers.base import BaseTrigger, TriggerEvent
-from google.api_core.exceptions import GoogleAPIError
-from google.cloud.storage_transfer_v1.types import TransferOperation
 
 
 class CloudStorageTransferServiceCreateJobsTrigger(BaseTrigger):

airflow/providers/google/cloud/triggers/dataflow.py

@@ -22,8 +22,6 @@ from collections.abc import Sequence
 from functools import cached_property
 from typing import TYPE_CHECKING, Any
 
-from airflow.providers.google.cloud.hooks.dataflow import AsyncDataflowHook, DataflowJobStatus
-from airflow.triggers.base import BaseTrigger, TriggerEvent
 from google.cloud.dataflow_v1beta3 import JobState
 from google.cloud.dataflow_v1beta3.types import (
     AutoscalingEvent,
@@ -34,6 +32,9 @@ from google.cloud.dataflow_v1beta3.types import (
     MetricUpdate,
 )
 
+from airflow.providers.google.cloud.hooks.dataflow import AsyncDataflowHook, DataflowJobStatus
+from airflow.triggers.base import BaseTrigger, TriggerEvent
+
 if TYPE_CHECKING:
     from google.cloud.dataflow_v1beta3.services.messages_v1_beta3.pagers import ListJobMessagesAsyncPager
 

airflow/providers/google/cloud/triggers/dataplex.py

@@ -22,9 +22,10 @@ from __future__ import annotations
 import asyncio
 from collections.abc import AsyncIterator, Sequence
 
+from google.cloud.dataplex_v1.types import DataScanJob
+
 from airflow.providers.google.cloud.hooks.dataplex import DataplexAsyncHook
 from airflow.triggers.base import BaseTrigger, TriggerEvent
-from google.cloud.dataplex_v1.types import DataScanJob
 
 
 class DataplexDataQualityJobTrigger(BaseTrigger):

airflow/providers/google/cloud/triggers/dataproc.py

@@ -25,6 +25,9 @@ import time
 from collections.abc import AsyncIterator, Sequence
 from typing import TYPE_CHECKING, Any
 
+from google.api_core.exceptions import NotFound
+from google.cloud.dataproc_v1 import Batch, Cluster, ClusterStatus, JobStatus
+
 from airflow.exceptions import AirflowException
 from airflow.models.taskinstance import TaskInstance
 from airflow.providers.google.cloud.hooks.dataproc import DataprocAsyncHook, DataprocHook
@@ -33,8 +36,6 @@ from airflow.providers.google.common.hooks.base_google import PROVIDE_PROJECT_ID
 from airflow.triggers.base import BaseTrigger, TriggerEvent
 from airflow.utils.session import provide_session
 from airflow.utils.state import TaskInstanceState
-from google.api_core.exceptions import NotFound
-from google.cloud.dataproc_v1 import Batch, Cluster, ClusterStatus, JobStatus
 
 if TYPE_CHECKING:
     from sqlalchemy.orm.session import Session

airflow/providers/google/cloud/triggers/kubernetes_engine.py

@@ -23,6 +23,7 @@ from collections.abc import AsyncIterator, Sequence
 from functools import cached_property
 from typing import TYPE_CHECKING, Any
 
+from google.cloud.container_v1.types import Operation
 from packaging.version import parse as parse_version
 
 from airflow.exceptions import AirflowException, AirflowProviderDeprecationWarning
@@ -36,7 +37,6 @@ from airflow.providers.google.cloud.hooks.kubernetes_engine import (
 )
 from airflow.providers_manager import ProvidersManager
 from airflow.triggers.base import BaseTrigger, TriggerEvent
-from google.cloud.container_v1.types import Operation
 
 if TYPE_CHECKING:
     from datetime import datetime

airflow/providers/google/cloud/triggers/pubsub.py

@@ -23,9 +23,10 @@ from collections.abc import AsyncIterator, Sequence
 from functools import cached_property
 from typing import Any
 
+from google.cloud.pubsub_v1.types import ReceivedMessage
+
 from airflow.providers.google.cloud.hooks.pubsub import PubSubAsyncHook
 from airflow.triggers.base import BaseTrigger, TriggerEvent
-from google.cloud.pubsub_v1.types import ReceivedMessage
 
 
 class PubsubPullTrigger(BaseTrigger):

airflow/providers/google/cloud/triggers/vertex_ai.py

@@ -20,6 +20,14 @@ from collections.abc import AsyncIterator, Sequence
 from functools import cached_property
 from typing import TYPE_CHECKING, Any
 
+from google.cloud.aiplatform_v1 import (
+    BatchPredictionJob,
+    HyperparameterTuningJob,
+    JobState,
+    PipelineState,
+    types,
+)
+
 from airflow.exceptions import AirflowException
 from airflow.providers.google.cloud.hooks.vertex_ai.batch_prediction_job import BatchPredictionJobAsyncHook
 from airflow.providers.google.cloud.hooks.vertex_ai.custom_job import CustomJobAsyncHook
@@ -28,13 +36,6 @@ from airflow.providers.google.cloud.hooks.vertex_ai.hyperparameter_tuning_job im
 )
 from airflow.providers.google.cloud.hooks.vertex_ai.pipeline_job import PipelineJobAsyncHook
 from airflow.triggers.base import BaseTrigger, TriggerEvent
-from google.cloud.aiplatform_v1 import (
-    BatchPredictionJob,
-    HyperparameterTuningJob,
-    JobState,
-    PipelineState,
-    types,
-)
 
 if TYPE_CHECKING:
     from proto import Message

airflow/providers/google/cloud/utils/credentials_provider.py

@@ -29,6 +29,10 @@ from urllib.parse import urlencode
 
 import google.auth
 import google.oauth2.service_account
+from google.auth import impersonated_credentials  # type: ignore[attr-defined]
+from google.auth.credentials import AnonymousCredentials, Credentials
+from google.auth.environment_vars import CREDENTIALS, LEGACY_PROJECT, PROJECT
+
 from airflow.exceptions import AirflowException
 from airflow.providers.google.cloud._internal_client.secret_manager_client import _SecretManagerClient
 from airflow.providers.google.cloud.utils.external_token_supplier import (
@@ -36,9 +40,6 @@ from airflow.providers.google.cloud.utils.external_token_supplier import (
 )
 from airflow.utils.log.logging_mixin import LoggingMixin
 from airflow.utils.process_utils import patch_environ
-from google.auth import impersonated_credentials  # type: ignore[attr-defined]
-from google.auth.credentials import AnonymousCredentials, Credentials
-from google.auth.environment_vars import CREDENTIALS, LEGACY_PROJECT, PROJECT
 
 log = logging.getLogger(__name__)
 

airflow/providers/google/cloud/utils/external_token_supplier.py

@@ -22,7 +22,6 @@ from functools import wraps
 from typing import TYPE_CHECKING, Any
 
 import requests
-
 from google.auth.exceptions import RefreshError
 from google.auth.identity_pool import SubjectTokenSupplier