Remove check for run_update.end_time as it is guaranteed to have a …

…time zone
zenml-io · Dec 16, 2024 · 5fbb998 · 5fbb998
1 parent 515ccfb
commit 5fbb998
Showing 1 changed file with 0 additions and 4 deletions.
diff --git 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