from nodes.base_input import BaseInput

from ...impl.ncnn.model import NcnnModelWrapper


class NcnnModelInput(BaseInput):
    """Input for NcnnModel"""

    def __init__(self, label: str = "Model"):
        super().__init__("NcnnNetwork", label)
        self.associated_type = NcnnModelWrapper