fix

pola-rs · Oct 8, 2024 · 027f115 · 027f115
1 parent 8ae6f94
commit 027f115
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/crates/polars-plan/src/plans/conversion/expr_expansion.rs b/crates/polars-plan/src/plans/conversion/expr_expansion.rs
@@ -938,7 +938,7 @@ pub(super) fn expand_selector(
         let selected = schema
             .iter_fields()
             .map(|field| field.name().clone())
-            .filter(|field_name| column_names.contains(&field_name))
+            .filter(|field_name| column_names.contains(field_name))
             .collect();
 
         Ok(selected)

diff --git a/py-polars/tests/unit/test_selectors.py b/py-polars/tests/unit/test_selectors.py
@@ -819,6 +819,6 @@ def test_selector_list_of_lists_18499() -> None:
 def test_invalid_selector() -> None:
     df = pl.DataFrame(data={"x": [1, 2], "z": ["a", "b"]})
     with pytest.raises(InvalidOperationError, match="invalid selector expression"):
-        df.drop(pl.col("x", "z") + 2)
+        df.drop(pl.col("x", "z") + 2)  # type: ignore[arg-type]
     with pytest.raises(InvalidOperationError, match="invalid selector expression"):
-        df.drop(pl.col("x") + 2)
+        df.drop(pl.col("x") + 2)  # type: ignore[arg-type]