keras/src/ops/numpy.py/Argpartition/__init__
class Argpartition:    def __init__(self, kth, axis=-1):
        super().__init__()
        if not isinstance(kth, int):
            raise ValueError("kth must be an integer. Received:" f"kth = {kth}")
        self.kth = kth
        self.axis = axis