def METHOD_NAME(self, X):
    result = self._wrapped_model.METHOD_NAME(X)
    try:
        if hasattr(self, "column_names") and len(self.column_names) == len(
            self._wrapped_model.cols_to_keep_final_
        ):
            self.column_names = [
                self.column_names[i]
                for i in self._wrapped_model.cols_to_keep_final_
            ]
        if hasattr(self, "column_dtypes") and len(self.column_dtypes) == len(
            self._wrapped_model.cols_to_keep_final_
        ):
            self.column_dtypes = [
                self.column_dtypes[i]
                for i in self._wrapped_model.cols_to_keep_final_
            ]
    except Exception:  # nosec
        pass
    return result