fix conflation of str/collection

pola-rs · Aug 2, 2023 · fa82d3b · fa82d3b
1 parent e4f7f1f
commit fa82d3b
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/py-polars/polars/lazyframe/frame.py b/py-polars/polars/lazyframe/frame.py
@@ -3371,7 +3371,11 @@ def drop(
 
         """
         input_cols: list[str | SelectorType] = [
-            *(columns if isinstance(columns, Collection) else [columns]),  # type: ignore[list-item]
+            *(
+                columns
+                if isinstance(columns, Collection) and not isinstance(columns, str)
+                else [columns]  # type: ignore[list-item]
+            ),
             *more_columns,
         ]
         drop_cols: list[str] = []