FIX correction of building a list instead of squeezing when index is …

…built from 1 element
rapidsai · Feb 7, 2024 · 4d64ea8 · 4d64ea8
1 parent 287e8d4
commit 4d64ea8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/cuml/preprocessing/encoders.py b/python/cuml/preprocessing/encoders.py
@@ -498,7 +498,7 @@ def inverse_transform(self, X):
                 dropped_class_idx = Series(self.drop_idx_[feature])
                 dropped_class_mask = Series(cats).isin(cats[dropped_class_idx])
                 if len(cats) == 1:
-                    inv = Series(Index(cp.squeeze(cats[0])).repeat(X.shape[0]))
+                    inv = Series(Index([cats[0]]).repeat(X.shape[0]))
                     result[feature] = inv
                     continue
                 cats = cats[~dropped_class_mask]