Source code for otx.config.device

# Copyright (C) 2023 Intel Corporation
# SPDX-License-Identifier: Apache-2.0

"""Config data type objects for device."""

from __future__ import annotations

from dataclasses import dataclass

from otx.types.device import DeviceType




@dataclass
class DeviceConfig:
    """Configuration class for the engine."""

    accelerator: DeviceType
    devices: int = 1