flax_model/alphafold3/structure/cpp/membership.pyi

import numpy


def isin(
    array: numpy.ndarray[numpy.int64],
    test_elements: set[int],
    invert: bool = ...,
) -> numpy.ndarray[bool]: ...