updated retry test

PrefectHQ · Apr 10, 2024 · eca4958 · eca4958
1 parent 4cae60d
commit eca4958
Showing 1 changed file with 1 addition and 3 deletions.
diff --git 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)