src/data/model_data.py

from dataclasses import dataclass
from src.interface import ModelInterface

@dataclass
class ModelData:
    name: str 
    model_class: ModelInterface