From baafa7126ccc26a3773ca6b091f2f1db2e24f66c Mon Sep 17 00:00:00 2001 From: Martynas Asipauskas Date: Wed, 4 Sep 2024 15:18:52 +0000 Subject: [PATCH] Keep last log line if no-logs (#216) --- third_party/airflow/armada/operators/armada.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/third_party/airflow/armada/operators/armada.py b/third_party/airflow/armada/operators/armada.py index 3f06b99252b..c7e51f1a536 100644 --- a/third_party/airflow/armada/operators/armada.py +++ b/third_party/airflow/armada/operators/armada.py @@ -340,9 +340,10 @@ def _check_job_status_and_fetch_logs(self) -> None: container=self.container_logs, since_time=self.job_context.last_log_time, ) - self.job_context = dataclasses.replace( - self.job_context, last_log_time=last_log_time - ) + if last_log_time: + self.job_context = dataclasses.replace( + self.job_context, last_log_time=last_log_time + ) except Exception as e: self.log.warning(f"Error fetching logs {e}")