def __post_init__()

in optimum_benchmark/launchers/config.py [0:0]

• 19 lines of code
• 8 McCabe index (conditional complexity)

    def __post_init__(self):
        if self.device_isolation and not is_nvidia_system() and not is_rocm_system():
            raise ValueError(
                "Device isolation is only supported on NVIDIA and ROCm systems. "
                "Please set `device_isolation` to False or make sure your drivers "
                "are correctly installed by running `nvidia-smi` or `rocm-smi`."
            )

        if self.device_isolation and self.device_isolation_action is None:
            LOGGER.warning(
                "Device isolation is enabled but no action is specified. "
                "Please set `device_isolation_action` to either `error`, `warn`, or `kill`. "
                "Defaulting to `warn`."
            )
            self.device_isolation_action = "warn"

        elif self.device_isolation and self.device_isolation_action not in {"error", "warn", "kill"}:
            raise ValueError(
                f"Unsupported device isolation action {self.device_isolation_action}. "
                "Please set `device_isolation_action` to either `error`, `warn`, or `kill`."
            )