feat(ecsoperator): Add execution role arn

openclimatefix · Mar 4, 2025 · cb15ab3 · cb15ab3
1 parent 945cdee
commit cb15ab3
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/airflow_dags/plugins/operators/ecs_run_task_operator.py b/src/airflow_dags/plugins/operators/ecs_run_task_operator.py
@@ -17,6 +17,7 @@
 AWS_ACCOUNT_ID: str = os.getenv("AWS_ACCOUNT_ID", "")
 ECS_SUBNET: str = os.getenv("ECS_SUBNET", "")
 ECS_SECURITY_GROUP: str = os.getenv("ECS_SECURITY_GROUP", "")
+ECS_EXECUTION_ROLE_ARN: str = os.getenv("ECS_EXECUTION_ROLE_ARN", "")
 AWS_REGION: str = os.getenv("AWS_REGION", "eu-west-1")
 
 @dataclasses.dataclass
@@ -109,6 +110,7 @@ def setup_operator(self) -> EcsRegisterTaskDefinitionOperator:
                 "cpu": str(self.container_cpu),
                 "memory": str(self.container_memory),
                 "requiresCompatibilities": ["FARGATE"],
+                "executionRoleArn": ECS_EXECUTION_ROLE_ARN,
                 "tags": [
                     {"key": "name", "value": self.name},
                     {"key": "environment", "value": ENV},