Trixy7/trixy_core/trainer/enums.py

# -*- coding: utf-8 -*-
"""
Enums fuer das Trainer-Framework.
"""

from enum import Enum


class TrainingState(Enum):
    """Status eines Trainings-Durchlaufs."""
    IDLE = "idle"
    VALIDATING = "validating"
    STARTING = "starting"
    TRAINING = "training"
    PAUSED = "paused"
    STOPPING = "stopping"
    COMPLETED = "completed"
    FAILED = "failed"


class TrainerType(Enum):
    """Herkunft eines Trainers."""
    CORE = "core"
    PLUGIN = "plugin"