Add Active Stack as part of configuration YAML

docs/book/user-guide/advanced-guide/pipelining-features/configure-steps-pipelines.md

@@ -132,6 +132,7 @@ python run.py
 <summary>An example of a generated YAML configuration template</summary>
 
 ```yaml
+active_stack: Optional[str]
 build: Union[PipelineBuildBase, UUID, NoneType]
 enable_artifact_metadata: Optional[bool]
 enable_artifact_visualization: Optional[bool]
@@ -330,6 +331,12 @@ These are boolean flags for various configurations:
 * `enable_cache`: Utilize [caching](../../starter-guide/cache-previous-executions.md) or not.
 * `enable_step_logs`: Enable tracking [step logs](managing-steps.md#enable-or-disable-logs-storing).
 
+### `active_stack` name or ID
+
+The name of the UUID of the `active stack` to use for this
+pipeline. If specified the active stack is set for the duration of the pipeline execution and restored upon 
+completion. If not specified, the current active stack is used.
+
 ### `build` ID
 
 The UUID of the [`build`](../infrastructure-management/containerize-your-pipeline.md) to use for this pipeline. If specified, Docker image building is skipped for remote orchestrators, and the Docker image specified in this build is used.

src/zenml/new/pipelines/pipeline.py

@@ -1580,7 +1580,7 @@ def _prepare_if_possible(self) -> None:
     def _update_stack_from_config(
         self, run_configuration: PipelineRunConfiguration
     ) -> None:
-        """Activate the stack from the pipeline run configuation if one is given.
+        """Activate the stack from the pipeline run configuration if one is given.
 
         Args:
             run_configuration: The run configuration for this pipeline.

src/zenml/stack/utils.py

@@ -149,16 +149,24 @@
         """Constructor for stack_context saves active stack."""
         self._default_stack = Client().active_stack
 
-    def __enter__(self):
+    def __enter__(self) -> None:
         """Enters in the stack context."""
         pass
 
     def __exit__(
         self,
-        exc_type: Optional[Type[BaseException]],
-        exc_value: Optional[BaseException],
+        exception_type: Optional[Type[BaseException]],
+        exception_value: Optional[BaseException],
         traceback: Optional[TracebackType],
     ) -> None:
-        """Restores the original active stack."""
+        """Get a stack component flavor by name and type from a ZenStore.
+
+        Args:
+            exception_type: Type of the execption that was raised.
+            None if no execption.
+            exception_value: Type of exception that was raised.
+            e.g., divide_by_zero error. None if no exception.
+            traceback: Traceback report. None if no excpetion.
+        """
         if self._default_stack.id != Client().active_stack:
             Client().activate_stack(self._default_stack.id)