diff --git a/cu_cat/_table_vectorizer.py b/cu_cat/_table_vectorizer.py index 3f843ba1f..3090d66df 100644 --- a/cu_cat/_table_vectorizer.py +++ b/cu_cat/_table_vectorizer.py @@ -821,11 +821,11 @@ def get_feature_names_out(self, input_features=None) -> List[str]: all_trans_feature_names.extend(cols) continue try: - ct_feature_names = super().get_feature_names_out() + trans_feature_names = super().get_feature_names_out() except: pass try: - ct_feature_names = super().get_feature_names() + trans_feature_names = super().get_feature_names() except: pass all_trans_feature_names.extend(trans_feature_names)