Skip to content

Commit

Permalink
[pre-commit.ci] auto fixes from pre-commit.com hooks
Browse files Browse the repository at this point in the history
for more information, see https://pre-commit.ci
  • Loading branch information
pre-commit-ci[bot] committed Mar 19, 2024
1 parent 1d33b50 commit fdfdc33
Show file tree
Hide file tree
Showing 3 changed files with 9 additions and 7 deletions.
4 changes: 2 additions & 2 deletions astronomer/providers/apache/hive/example_dags/example_hive.py
Original file line number Diff line number Diff line change
Expand Up @@ -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)
Expand Down
4 changes: 2 additions & 2 deletions astronomer/providers/apache/livy/example_dags/example_livy.py
Original file line number Diff line number Diff line change
Expand Up @@ -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)
Expand Down
8 changes: 5 additions & 3 deletions astronomer/providers/sftp/example_dags/example_sftp.py
Original file line number Diff line number Diff line change
Expand Up @@ -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)
Expand Down Expand Up @@ -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(
Expand Down

0 comments on commit fdfdc33

Please sign in to comment.