From 5fbb998a1c4cc39458b6a80c88351669336d760a Mon Sep 17 00:00:00 2001 From: Nils Uhrberg Date: Mon, 16 Dec 2024 16:27:56 +0100 Subject: [PATCH] Remove check for `run_update.end_time` as it is guaranteed to have a time zone --- src/zenml/zen_stores/sql_zen_store.py | 4 ---- 1 file changed, 4 deletions(-) diff --git a/src/zenml/zen_stores/sql_zen_store.py b/src/zenml/zen_stores/sql_zen_store.py index 06071fd6cc..9c535e573c 100644 --- a/src/zenml/zen_stores/sql_zen_store.py +++ b/src/zenml/zen_stores/sql_zen_store.py @@ -8640,10 +8640,6 @@ def _update_pipeline_run_status( tzinfo=timezone.utc ) ) - if run_update.end_time.tzinfo is None: - run_update.end_time = run_update.end_time.replace( - tzinfo=timezone.utc - ) duration_time = ( run_update.end_time - pipeline_run.start_time