Class PythonOperatorPipeline

dali/test/python/test_dali_tf_plugin_run.py:163–169

class PythonOperatorPipeline(Pipeline):
    def __init__(self):
        super().__init__(1, 1, 0, 0)
        self.python_op = ops.PythonFunction(function=lambda: np.zeros((3, 3, 3)))

    def define_graph(self):
        return self.python_op()


@raises(