Class ArgCb

dali/test/python/sequences_test_utils.py:60–91

Describes a callback to be used as a per-sample/per-frame argument to the operator. ---------- name : Union[str, int] String with the name of a named argument of the operator or an int if the data should be passed as a positional input. cb : Callable[[SampleDesc

class ArgCb:
    """
    Describes a callback to be used as a per-sample/per-frame argument to the operator.
    ----------
    name : Union[str, int]
        String with the name of a named argument of the operator or an int if the data
             should be passed as a positional input.
    cb : Callable[[SampleDesc], np.ndarray]
        Callback that based on the SampleDesc instance produces a single parameter for
             specific sample/frame.
    is_per_frame : bool
        Flag if the cb should be run for every sample (sequence) or for every frame.
             In the latter case, the argument is passed wrapped
             in per-frame call to the operator.
    dest_device : str
        Controls whether the produced data should be passed to the operator in cpu or gpu memory.
        If set to "gpu", the copy to gpu is added in the pipeline.
             Applicable only to positional inputs.
    """

    def __init__(
        self,
        name: Union[str, int],
        cb: Callable[[SampleDesc], np.ndarray],
        is_per_frame: bool,
        dest_device: str = "cpu",
    ):
        self.desc = ArgDesc(name, "F" if is_per_frame else "", dest_device)
        self.cb = cb

    def __repr__(self):
        return "ArgCb{}".format((self.cb, self.desc))


@dataclass