Class ArgDesc

dali/test/python/sequences_test_utils.py:43–57

@dataclass
class ArgDesc:
    name: Union[str, int]
    expandable_prefix: str
    dest_device: str
    layout: Optional[str] = None

    def __post_init__(self):
        assert (
            self.is_positional_arg or self.dest_device != "gpu"
        ), "Named arguments on GPU are not supported"
        assert not self.layout or self.layout.startswith(self.expandable_prefix)

    @property
    def is_positional_arg(self):
        return isinstance(self.name, int)


class ArgCb: