27 lines
1.1 KiB
Python
27 lines
1.1 KiB
Python
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'
|