def __init__( self, keys: KeysCollection, select_labels: Union[Sequence[int], int], merge_channels: bool = False, ) -> None: # pytype: disable=annotation-type-mismatch # pytype bug with bool super().__init__(keys) self.converter = LabelToMask(select_labels, merge_channels)
def __init__(self, keys: KeysCollection, select_labels: Union[Sequence[int], int], merge_channels: bool = False) -> None: super().__init__(keys) self.converter = LabelToMask(select_labels, merge_channels)