from enum import StrEnum


class ExperimentStatus(StrEnum):
    """
    Status values for experiment run lifecycle.

    This enum defines all possible status values that an experiment run can have
    throughout its lifecycle, from initialization through training, model saving,
    and cleanup. These statuses are used to track progress and identify failures
    in the training pipeline.

    The status values follow the naming convention from the original Mage pipeline
    to maintain compatibility with existing database records and monitoring systems.

    Attributes:
        ORCHESTRATOR_WAITING_PROC: Initial status indicating experiment is registered and waiting for processing.
        ORCHESTRATOR_VALIDATION_ERROR: Error in the parameters validation.
        TRAINING_SUCCESS: Training completed successfully with model and metrics calculated.
        TRAINING_ERROR: Training failed due to data issues, model errors, or other exceptions.
    """

    ORCHESTRATOR_VALIDATION_ERROR = 'ORCHESTRATOR_VALIDATION_ERROR'
    ORCHESTRATOR_WAITING_PROC = 'ORCHESTRATOR_WAITING_PROC'
    TRAINING_SUCCESS = 'TRAINING_SUCCESS'
    TRAINING_ERROR = 'TRAINING_ERROR'