<filename>ml_model_evaluation/models/model_registry.py
from enum import Enum


class ModelRegistry(Enum):
    mlflow = 1
    s3 = 2

<filename>ml_model_evaluation/models/model_registry.py
from enum import Enum


class ModelRegistry(Enum):
    mlflow = 1
    s3 = 2
