Device

This class represents a hardware device configuration, specifying the device type, quantity, and optional partitioning details. It ensures valid resource allocation by enforcing a minimum quantity and supports various device classes and specific hardware models like GPUs or TPUs.

Attributes

Attribute	Type	Description
quantity	int	The number of devices of this type, which must be an integer of at least 1.
device_class	DeviceClass	The category or classification of the hardware device.
device	`str	None` = null
partition	`str	None` = null

---

Constructor

Signature

def Device(
    quantity: int,
    device_class: DeviceClass,
    device: str | None = None,
    partition: str | None = None
) -> null

Parameters

Name	Type	Description
quantity	int	The number of devices of this type.
device_class	DeviceClass	The classification category of the device.
device	`str	None` = None
partition	`str	None` = None

---