import enum

__all__ = ["Event"]


class Event(enum.Enum):
    BATCH_STARTED = "BATCH STARTED"
    BATCH_ENDED = "BATCH ENDED"
    EPOCH_STARTED = "EPOCH STARTED"
    EPOCH_ENDED = "EPOCH ENDED"
    TESTING_STARTED = "TESTING STARTED"
    TESTING_ENDED = "TESTING ENDED"
    TRAINING_STARTED = "TRAINING STARTED"
    TRAINING_ENDED = "TRAINING ENDED"
    # TUNING_STARTED = "TUNING STARTED"
    # TUNING_ENDED = "TUNING ENDED"
