code/self_healing_finetuning/n21/enums.py

from __future__ import annotations

from enum import StrEnum


class Provider(StrEnum):
    HF_MANAGED = "hf_managed"
    POD_CLOUD = "pod_cloud"
    LOCAL_NATIVE = "local_native"


class Environment(StrEnum):
    DEV = "dev"
    STAGE = "stage"
    PROD = "prod"


class LifecycleStage(StrEnum):
    OBSERVE = "observe"
    DIAGNOSE = "diagnose"
    REPAIR = "repair"
    RETRAIN = "retrain"
    REDEPLOY = "redeploy"
    REEVAL = "reeval"
    PROMOTE = "promote"
    ROLLBACK = "rollback"


class ErrorCode(StrEnum):
    E_AUTH = "E_AUTH"
    E_CONFIG = "E_CONFIG"
    E_CAPACITY = "E_CAPACITY"
    E_STORAGE_IO = "E_STORAGE_IO"
    E_DATASET_SCHEMA = "E_DATASET_SCHEMA"
    E_LICENSE_BLOCK = "E_LICENSE_BLOCK"
    E_PII_BLOCK = "E_PII_BLOCK"
    E_MNPI_BLOCK = "E_MNPI_BLOCK"
    E_TRAIN_OOM = "E_TRAIN_OOM"
    E_TRAIN_DIVERGENCE = "E_TRAIN_DIVERGENCE"
    E_PROVIDER_TIMEOUT = "E_PROVIDER_TIMEOUT"
    E_RATE_LIMIT = "E_RATE_LIMIT"
    E_INFERENCE_UNHEALTHY = "E_INFERENCE_UNHEALTHY"
    E_EVAL_REGRESSION = "E_EVAL_REGRESSION"
    E_PROMOTION_BLOCK = "E_PROMOTION_BLOCK"