from typing import List, Tuple
import numpy as np


cdef class InferenceEngine:
    cdef public int batch_size
    cdef tuple get_input_shape(self)
    cdef int get_batch_size(self)
    cdef run(self, input_data)