From fde3d3e11e574124774229a4d09c0659ca84e757 Mon Sep 17 00:00:00 2001 From: Daniel Date: Fri, 29 Dec 2023 11:09:20 +0800 Subject: [PATCH] better sklearn ver test + table tests --- cu_cat/_table_vectorizer.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) 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)