Source code for satorbis_kit.ml_lifecycle.exceptions.mlflow

from __future__ import annotations

from typing import Optional




class MLFlowError(Exception):
    """Base exception for all ml_lifecycle operations."""





class MLFlowConnectionError(MLFlowError):
    """Raised when connection to the MLflow tracking server or registry fails."""

    def __init__(self, message: str, cause: Optional[Exception] = None) -> None:
        super().__init__(message)
        self.cause = cause





class ExperimentError(MLFlowError):
    """Raised when experiment operations fail."""





class RunError(MLFlowError):
    """Raised when run operations fail."""





class RegistryError(MLFlowError):
    """Raised when model registry operations fail."""





class ModelCardError(MLFlowError):
    """Raised when model card operations fail."""





class ArtifactError(MLFlowError):
    """Raised when artifact operations fail."""