Skip to content

Commit

Permalink
Fix task name to get cluster IP using xcom (#1507)
Browse files Browse the repository at this point in the history
We were not using correct task for cluster IP retrieval. Updated task name correctly.
  • Loading branch information
vatsrahul1001 authored Apr 12, 2024
1 parent cdef29f commit a97ef0e
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion astronomer/providers/sftp/example_dags/example_sftp.py
Original file line number Diff line number Diff line change
Expand Up @@ -167,7 +167,7 @@ def revoke_inbound_rules(task_instance: TaskInstance) -> None:
FromPort=22,
ToPort=22,
GroupId=task_instance.xcom_pull(
key="cluster_response_master_security_group", task_ids=["describe_created_cluster"]
key="instance_response_master_security_group", task_ids=["create_ec2_instance"]
)[0],
IpProtocol="tcp",
)
Expand Down

0 comments on commit a97ef0e

Please sign in to comment.