def METHOD_NAME(self, X):
    if isinstance(X, pd.DataFrame):
        keep_indices = self._wrapped_model.get_support(indices=True)
        keep_columns = [X.columns[i] for i in keep_indices]
        result = X[keep_columns]
    else:
        result = self._wrapped_model.METHOD_NAME(X)
    return result