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