optimum_benchmark/scenarios/config.py

from abc import ABC
from dataclasses import dataclass
from logging import getLogger
from typing import TypeVar

LOGGER = getLogger("benchmark")


@dataclass
class ScenarioConfig(ABC):
    name: str
    _target_: str

    def __post_init__(self):
        pass


ScenarioConfigT = TypeVar("ScenarioConfigT", bound=ScenarioConfig)