diff --git a/airflow/providers/amazon/aws/hooks/logs.py b/airflow/providers/amazon/aws/hooks/logs.py index bb22ae13d7337..fb12943f65fa8 100644 --- a/airflow/providers/amazon/aws/hooks/logs.py +++ b/airflow/providers/amazon/aws/hooks/logs.py @@ -71,9 +71,7 @@ def get_log_events( | 'ingestionTime' (int): The time in milliseconds the event was ingested. """ next_token = None - - event_count = 1 - while event_count > 0: + while True: if next_token is not None: token_arg: Optional[Dict[str, str]] = {'nextToken': next_token} else: @@ -99,7 +97,7 @@ def get_log_events( yield from events - if 'nextForwardToken' in response: + if next_token != response['nextForwardToken']: next_token = response['nextForwardToken'] else: return