From e0cacb5466d865ddedff6099c3dc953593adb570 Mon Sep 17 00:00:00 2001 From: Ankit Chaurasia <8670962+sunank200@users.noreply.github.com> Date: Thu, 25 Apr 2024 16:38:08 +0545 Subject: [PATCH] Fix the PR comments --- airflow/providers/google/cloud/triggers/dataproc.py | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/airflow/providers/google/cloud/triggers/dataproc.py b/airflow/providers/google/cloud/triggers/dataproc.py index 0b0e0c60d44776..32b536a2ecaa37 100644 --- a/airflow/providers/google/cloud/triggers/dataproc.py +++ b/airflow/providers/google/cloud/triggers/dataproc.py @@ -181,6 +181,7 @@ async def run(self) -> AsyncIterator[TriggerEvent]: } ) return + self.log.info("Current state is %s", state) self.log.info("Sleeping for %s seconds.", self.polling_interval_seconds) await asyncio.sleep(self.polling_interval_seconds) except asyncio.CancelledError: @@ -205,7 +206,7 @@ async def fetch_cluster(self) -> Cluster: project_id=self.project_id, region=self.region, cluster_name=self.cluster_name ) - async def delete_when_error_occurred(self, cluster: Cluster): + async def delete_when_error_occurred(self, cluster: Cluster) -> None: """ Delete the cluster on error. @@ -218,9 +219,7 @@ async def delete_when_error_occurred(self, cluster: Cluster): ) self.log.info("Cluster %s has been deleted.", self.cluster_name) else: - self.log.info( - "Cluster %s is not be deleted as delete_on_error is set to False.", self.cluster_name - ) + self.log.info("Cluster %s is not deleted as delete_on_error is set to False.", self.cluster_name) class DataprocBatchTrigger(DataprocBaseTrigger):