diff --git a/tests/system/providers/amazon/aws/example_batch.py b/tests/system/providers/amazon/aws/example_batch.py index 77c5d21997364..26059a2c2fb8d 100644 --- a/tests/system/providers/amazon/aws/example_batch.py +++ b/tests/system/providers/amazon/aws/example_batch.py @@ -30,6 +30,7 @@ BatchJobQueueSensor, BatchSensor, ) +from airflow.providers.amazon.aws.operators.ecs import EcsDeregisterTaskDefinitionOperator from airflow.utils.trigger_rule import TriggerRule from tests.system.providers.amazon.aws.utils import ( ENV_ID_KEY, @@ -240,6 +241,13 @@ def delete_job_queue(job_queue_name): poke_interval=10, ) + deregister_task_definition = EcsDeregisterTaskDefinitionOperator( + task_id="deregister_task", + task_definition=f"{batch_job_definition_name}:1", + trigger_rule=TriggerRule.ALL_DONE, + ) + + log_cleanup = prune_logs( [ # Format: ('log group name', 'log stream prefix')