Manage Stack Component States

Some stack components come with built-in daemons for connecting to the underlying remote infrastructure. These stack components expose functionality for provisioning, deprovisioning, starting, or stopping the corresponding daemons.
See the advanced section on Services for more information on daemons.
For such components, you can manage the daemon state using the zenml <STACK_COMPONENT> up and zenml <STACK_COMPONENT> down commands. Alternatively, you can also use zenml stack up or zenml stack down to manage the state of your entire stack:
zenml stack up  # Provision and start all stack components
zenml metadata-store up  # Provision and start the metadata store only
​
zenml stack down  # Stop all stack components
zenml metadata-store down  # Stop the metadata store only
​
zenml stack down --force  # Stop and deprovision all stack components
zenml metadata-store down --force  # Stop and deprovision the metadata store only
Defining States of Custom Components
By default, each stack component is assumed to be in a provisioned and running state right after creation. However, if you want to write a custom component and have fine-grained control over its state, you can overwrite the following properties and methods of the StackComponent base interface to configure the component according to your needs:
from abc import ABC
from pydantic import BaseModel
​
​
class StackComponent(BaseModel, ABC):
    """Abstract class for all components of a ZenML stack."""
    ...
    
    @property
    def is_provisioned(self) -> bool:
        """If the component provisioned resources to run."""
        return True
​
    @property
    def is_running(self) -> bool:
        """If the component is running."""
        return True
​
    def provision(self) -> None:
        """Provisions resources to run the component."""
​
    def deprovision(self) -> None:
        """Deprovisions all resources of the component."""
        
    def resume(self) -> None:
        """Resumes the provisioned resources of the component."""
​
    def suspend(self) -> None:
        """Suspends the provisioned resources of the component."""
​
    ...

Write Custom Stack Component Flavors

Pass Custom Data Types through Steps

Last modified 7d ago

Copy link