def METHOD_NAME(self, X: pd.DataFrame, **kwargs) -> np.ndarray:
    X = self.preprocess(X, **kwargs)
    pred_labels, pred_probs = self.model.METHOD_NAME(X)
    y_pred = np.array(
        [self._label_inv_map[labels[0]] for labels in pred_labels],
        dtype=self._label_dtype,
    )
    return y_pred