From 5ff7a81eb100540c7ebfe5fcd19c7ef229a2fee5 Mon Sep 17 00:00:00 2001 From: vatsrahul1001 Date: Fri, 12 Apr 2024 08:16:22 +0530 Subject: [PATCH 1/2] fixing cluster ip xcom retrival --- astronomer/providers/sftp/example_dags/example_sftp.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/astronomer/providers/sftp/example_dags/example_sftp.py b/astronomer/providers/sftp/example_dags/example_sftp.py index 9af9eedcb..62babc649 100644 --- a/astronomer/providers/sftp/example_dags/example_sftp.py +++ b/astronomer/providers/sftp/example_dags/example_sftp.py @@ -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="create_ec2_instance", task_ids=["instance_response_master_security_group"] )[0], IpProtocol="tcp", ) From 55487b2434e45f64ff3dc06a117666ff7bed52fd Mon Sep 17 00:00:00 2001 From: vatsrahul1001 Date: Fri, 12 Apr 2024 08:19:37 +0530 Subject: [PATCH 2/2] fix taskname --- astronomer/providers/sftp/example_dags/example_sftp.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/astronomer/providers/sftp/example_dags/example_sftp.py b/astronomer/providers/sftp/example_dags/example_sftp.py index 62babc649..8c2426b7a 100644 --- a/astronomer/providers/sftp/example_dags/example_sftp.py +++ b/astronomer/providers/sftp/example_dags/example_sftp.py @@ -167,7 +167,7 @@ def revoke_inbound_rules(task_instance: TaskInstance) -> None: FromPort=22, ToPort=22, GroupId=task_instance.xcom_pull( - key="create_ec2_instance", task_ids=["instance_response_master_security_group"] + key="instance_response_master_security_group", task_ids=["create_ec2_instance"] )[0], IpProtocol="tcp", )