From fdfdc33bdd94b525a2e1778dacabdeeff211bc85 Mon Sep 17 00:00:00 2001 From: "pre-commit-ci[bot]" <66853113+pre-commit-ci[bot]@users.noreply.github.com> Date: Tue, 19 Mar 2024 08:24:31 +0000 Subject: [PATCH] [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci --- .../providers/apache/hive/example_dags/example_hive.py | 4 ++-- .../providers/apache/livy/example_dags/example_livy.py | 4 ++-- astronomer/providers/sftp/example_dags/example_sftp.py | 8 +++++--- 3 files changed, 9 insertions(+), 7 deletions(-) diff --git a/astronomer/providers/apache/hive/example_dags/example_hive.py b/astronomer/providers/apache/hive/example_dags/example_hive.py index c6d8c50e4..20993a501 100644 --- a/astronomer/providers/apache/hive/example_dags/example_hive.py +++ b/astronomer/providers/apache/hive/example_dags/example_hive.py @@ -222,8 +222,8 @@ def revoke_inbound_rules(task_instance): CidrIp=ip_range, FromPort=22, GroupId=task_instance.xcom_pull( - key="cluster_response_master_security_group", task_ids=["describe_created_cluster"] - )[0], + key="cluster_response_master_security_group", task_ids=["describe_created_cluster"] + )[0], IpProtocol="tcp", ) logging.info("%s", response) diff --git a/astronomer/providers/apache/livy/example_dags/example_livy.py b/astronomer/providers/apache/livy/example_dags/example_livy.py index bad0fb420..7165f02d6 100644 --- a/astronomer/providers/apache/livy/example_dags/example_livy.py +++ b/astronomer/providers/apache/livy/example_dags/example_livy.py @@ -186,8 +186,8 @@ def revoke_inbound_rules(task_instance): FromPort=LIVY_OPERATOR_INGRESS_PORT, ToPort=LIVY_OPERATOR_INGRESS_PORT, GroupId=task_instance.xcom_pull( - key="cluster_response_master_security_group", task_ids=["describe_created_cluster"] - )[0], + key="cluster_response_master_security_group", task_ids=["describe_created_cluster"] + )[0], IpProtocol="tcp", ) logging.info("%s", response) diff --git a/astronomer/providers/sftp/example_dags/example_sftp.py b/astronomer/providers/sftp/example_dags/example_sftp.py index a5be711a4..ad7b37020 100644 --- a/astronomer/providers/sftp/example_dags/example_sftp.py +++ b/astronomer/providers/sftp/example_dags/example_sftp.py @@ -169,8 +169,8 @@ def revoke_inbound_rules(task_instance): FromPort=22, ToPort=22, GroupId=task_instance.xcom_pull( - key="cluster_response_master_security_group", task_ids=["describe_created_cluster"] - )[0], + key="cluster_response_master_security_group", task_ids=["describe_created_cluster"] + )[0], IpProtocol="tcp", ) logging.info("%s", response) @@ -299,7 +299,9 @@ def check_dag_status(**kwargs: Any) -> None: ) revoke_inbound_rules = PythonOperator( - task_id="revoke_inbound_rules", trigger_rule=TriggerRule.ALL_DONE, python_callable=revoke_inbound_rules + task_id="revoke_inbound_rules", + trigger_rule=TriggerRule.ALL_DONE, + python_callable=revoke_inbound_rules, ) dag_final_status = PythonOperator(