33 lines
1.4 KiB
Python
33 lines
1.4 KiB
Python
from enum import Enum
|
|
|
|
|
|
class ExperimentStatus(str, Enum):
|
|
"""
|
|
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:
|
|
MAGE_WAITING_PROC: Initial status indicating experiment is registered and waiting for processing.
|
|
TRAINING_SUCCESS: Training completed successfully with model and metrics calculated.
|
|
TRAINING_ERROR: Training failed due to data issues, model errors, or other exceptions.
|
|
MLFLOW_SENT: Model successfully saved to MLFlow.
|
|
MLFLOW_SEND_ERROR: Model saving to MLFlow failed due to connection or serialization errors.
|
|
FILE_DELETED: Cleanup completed successfully with all artifacts removed.
|
|
FILE_DELETE_ERROR: Cleanup failed due to file system or MinIO errors.
|
|
"""
|
|
|
|
MAGE_WAITING_PROC = 'MAGE_WAITING_PROC'
|
|
TRAINING_SUCCESS = 'TRAINING_SUCCESS'
|
|
TRAINING_ERROR = 'TRAINING_ERROR'
|
|
MLFLOW_SENT = 'MLFLOW_SENT'
|
|
MLFLOW_SEND_ERROR = 'MLFLOW_SEND_ERROR'
|
|
FILE_DELETED = 'FILE_DELETED'
|
|
FILE_DELETE_ERROR = 'FILE_DELETE_ERROR'
|