better sklearn ver test + table tests

graphistry · Dec 29, 2023 · fde3d3e · fde3d3e
1 parent c61bae8
commit fde3d3e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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)