fixing unit tests

Author: bcdurak

src/zenml/orchestrators/local/local_orchestrator.py

@@ -68,7 +68,7 @@ def submit_pipeline(
             Optional submission result.
 
         Raises:
-            RuntimeError: If the pipeline run fails.
+            Exception: If the pipeline run fails.
         """
         if deployment.schedule:
             logger.warning(
@@ -134,13 +134,11 @@ def submit_pipeline(
 
             try:
                 self.run_step(step=step)
-            except Exception as e:
+            except Exception:
                 failed_steps.append(step_name)
 
                 if execution_mode == ExecutionMode.FAIL_FAST:
-                    raise RuntimeError(
-                        f"Step {step_name} failed with error: {e}"
-                    )
+                    raise
 
         if failed_steps:
             raise RuntimeError(

src/zenml/orchestrators/local_docker/local_docker_orchestrator.py

@@ -123,7 +123,7 @@ def submit_pipeline(
             placeholder_run: An optional placeholder run for the deployment.
 
         Raises:
-            RuntimeError: If the pipeline run fails.
+            ContainerError: If the pipeline run fails.
 
         Returns:
             Optional submission result.
@@ -249,7 +249,7 @@ def submit_pipeline(
                 error_message = e.stderr.decode()
                 failed_steps.append(step_name)
                 if execution_mode == ExecutionMode.FAIL_FAST:
-                    raise RuntimeError(error_message)
+                    raise
                 else:
                     logger.error(error_message)