Skip to content

Commit 05440e2

Browse files
committed
fix col transf
1 parent e449190 commit 05440e2

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

sklearn/compose/_column_transformer.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1226,11 +1226,12 @@ def _hstack(self, Xs, *, n_samples):
12261226

12271227
def _sk_visual_block_(self):
12281228
# We can find remainder and its column only when it's fitted
1229+
12291230
if hasattr(self, "transformers_"):
12301231
transformers = [
12311232
transformer
1232-
for transformer in self.transformers_[:-1] # Exclude remainder
1233-
if not (transformer[0] == "remainder" and transformer[1] == "drop")
1233+
for transformer in self.transformers_ # Exclude remainder
1234+
if not (transformer[0] == "remainder")
12341235
]
12351236

12361237
# Add remainder back to fitted transformers if remainder is not drop

0 commit comments

Comments
 (0)