From eca49588ad3539f7089135e19cd043f56d324ee8 Mon Sep 17 00:00:00 2001 From: jeanluciano Date: Wed, 10 Apr 2024 13:49:34 -0500 Subject: [PATCH] updated retry test --- tests/workers/test_ecs_worker.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/tests/workers/test_ecs_worker.py b/tests/workers/test_ecs_worker.py index 0392d088..2e8a75a5 100644 --- a/tests/workers/test_ecs_worker.py +++ b/tests/workers/test_ecs_worker.py @@ -21,8 +21,6 @@ else: from pydantic import ValidationError -from tenacity import RetryError - from prefect_aws.credentials import _get_client_cached from prefect_aws.workers.ecs_worker import ( _TASK_DEFINITION_CACHE, @@ -2320,7 +2318,7 @@ async def test_retry_on_failed_task_start( }, ) - with pytest.raises(RetryError): + with pytest.raises(RuntimeError): async with ECSWorker(work_pool_name="test") as worker: await run_then_stop_task(worker, configuration, flow_run)