towhee.serve.triton.python_model_builder.OpPyModelBuilder

class towhee.serve.triton.python_model_builder.OpPyModelBuilder(task_name: str, op_name: str, op_init_args: Dict, input_annotations: List[Tuple[Any, Tuple]], output_annotations: List[Tuple[Any, Tuple]])

Bases: PyModelBuilder

Build python model from operator

Methods

build
gen_blank_lines
gen_class_name
gen_execute
gen_finalize
gen_imports
gen_initialize

__init__(task_name: str, op_name: str, op_init_args: Dict, input_annotations: List[Tuple[Any, Tuple]], output_annotations: List[Tuple[Any, Tuple]])