Method __init__

dali/test/python/operator_2/test_python_function.py:163–165

(self, batch_size, device_id, seed, image_dir, function)

class SinkTestPipeline(CommonPipeline):
    def __init__(self, batch_size, device_id, seed, image_dir, function):
        super().__init__(batch_size, 1, device_id, seed, image_dir)
        self.python_function = ops.PythonFunction(function=function, num_outputs=0)

    def define_graph(self):
        images, labels = self.load()