Call existing function for getting recent revision from family (#393)

PrefectHQ · Mar 12, 2024 · 1add0b9 · 1add0b9
1 parent 72c106f
commit 1add0b9
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/prefect_aws/workers/ecs_worker.py b/prefect_aws/workers/ecs_worker.py
@@ -769,7 +769,7 @@ def _get_or_register_task_definition(
         ):
             family_name = task_definition.get("family", ECS_DEFAULT_FAMILY)
             try:
-                task_definition_from_family = self._retrieve_task_definition_by_family(
+                task_definition_from_family = self._retrieve_task_definition(
                     logger, ecs_client, family_name
                 )
                 if task_definition_from_family and self._task_definitions_equal(
@@ -779,7 +779,7 @@ def _get_or_register_task_definition(
                         "taskDefinitionArn"
                     ]
             except Exception:
-                pass
+                cached_task_definition_arn = None
 
         if not cached_task_definition_arn:
             task_definition_arn = self._register_task_definition(