ml_pipeline/utils.py

from enum import Enum, auto


class Stage(Enum):
    TRAIN = auto()
    DEV = auto()
    TEST = auto()