Fix: step status check bug

Author: Json-Andriopoulos

src/zenml/orchestrators/step_launcher.py

@@ -370,14 +370,15 @@ def _bypass() -> None:
                     ):
                         # Handle as a non-failure as exception is a propagation of graceful termination.
                         publish_utils.publish_stopped_step_run(step_run.id)
+
                     else:
                         logger.error(
                             "Failed to run step `%s`: %s",
                             self._invocation_id,
                             e,
                         )
                         publish_utils.publish_failed_step_run(step_run.id)
-                        raise
+                    raise
             else:
                 logger.info(
                     f"Using cached version of step `{self._invocation_id}`."

src/zenml/zen_stores/schemas/pipeline_run_schemas.py

@@ -818,6 +818,7 @@ def is_placeholder_run(self) -> bool:
     @staticmethod
     def _step_in_progress(step_id: UUID, session: Session) -> bool:
         from zenml.steps.heartbeat import cached_is_heartbeat_unhealthy
+        from zenml.zen_stores.schemas.step_run_schemas import StepRunSchema
 
         step_run = session.execute(
             select(StepRunSchema).where(StepRunSchema.id == step_id)
@@ -899,18 +900,14 @@ def _check_if_run_in_progress(self) -> bool:
 
                         steps_statuses = {
                             name: ExecutionStatus(status)
-                            for name, status in step_run_statuses
+                            for _, name, status in step_run_statuses
                         }
 
                         for step_name, _ in step_dict.items():
                             if step_name in steps_to_skip:
                                 # failed steps downstream
                                 continue
 
-                            if steps_statuses[step_name].is_finished:
-                                # completed steps
-                                continue
-
                             if step_name not in steps_statuses:
                                 # steps that haven't started yet
                                 return True