Skip to content

Enums

zenml.enums

ArtifactStoreTypes (str, Enum)

All supported Artifact Store types.

Source code in zenml/enums.py
class ArtifactStoreTypes(str, Enum):
    """All supported Artifact Store types."""

    local = "local"
    gcp = "gcp"

    def __str__(self) -> str:
        """Returns the enum string value."""
        return self.value

ExecutionStatus (Enum)

Enum that represents the current status of a step or pipeline run.

Source code in zenml/enums.py
class ExecutionStatus(Enum):
    """Enum that represents the current status of a step or pipeline run."""

    FAILED = "failed"
    COMPLETED = "completed"
    RUNNING = "running"
    CACHED = "cached"

LoggingLevels (Enum)

Enum for logging levels.

Source code in zenml/enums.py
class LoggingLevels(Enum):
    """Enum for logging levels."""

    NOTSET = logging.NOTSET
    ERROR = logging.ERROR
    WARN = logging.WARN
    INFO = logging.INFO
    DEBUG = logging.DEBUG
    CRITICAL = logging.CRITICAL

MLMetadataTypes (str, Enum)

All supported ML Metadata types.

Source code in zenml/enums.py
class MLMetadataTypes(str, Enum):
    """All supported ML Metadata types."""

    sqlite = "sqlite"
    mysql = "mysql"
    kubeflow = "kubeflow"

    def __str__(self) -> str:
        """Returns the enum string value."""
        return self.value

OrchestratorTypes (str, Enum)

All supported Orchestrator types

Source code in zenml/enums.py
class OrchestratorTypes(str, Enum):
    """All supported Orchestrator types"""

    local = "local"
    airflow = "airflow"
    kubeflow = "kubeflow"

    def __str__(self) -> str:
        """Returns the enum string value."""
        return self.value

StackTypes (str, Enum)

All supported Stack types.

Source code in zenml/enums.py
class StackTypes(str, Enum):
    """All supported Stack types."""

    base = "base"