Update skrub/_table_vectorizer.py

Co-authored-by: Jérôme Dockès <[email protected]>
skrub-data · Nov 6, 2023 · 28e4c0d · 28e4c0d
1 parent c3f5636
commit 28e4c0d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/skrub/_table_vectorizer.py b/skrub/_table_vectorizer.py
@@ -150,8 +150,8 @@ def _replace_missing_in_cat_col(ser: pd.Series, value: str = "missing") -> pd.Se
     return ser
 
 
-def _clone_if_default(transformer, DEFAULT_TRANSFORMER):
-    return clone(transformer) if transformer is DEFAULT_TRANSFORMER else transformer
+def _clone_if_default(transformer, default_transformer):
+    return clone(transformer) if transformer is default_transformer else transformer
 
 
 def _clone_during_fit(transformer, remainder, n_jobs):