From fe2af1d0f652752621d2960efe0d3b73f2ec84ac Mon Sep 17 00:00:00 2001 From: Refael Ackermann Date: Sat, 27 Jul 2024 19:57:52 -0400 Subject: [PATCH] fix signature of frame.filter items can be ListLike --- pandas-stubs/core/frame.pyi | 2 +- tests/test_frame.py | 2 ++ 2 files changed, 3 insertions(+), 1 deletion(-) diff --git a/pandas-stubs/core/frame.pyi b/pandas-stubs/core/frame.pyi index 215f14711..20879b000 100644 --- a/pandas-stubs/core/frame.pyi +++ b/pandas-stubs/core/frame.pyi @@ -1711,7 +1711,7 @@ class DataFrame(NDFrame, OpsMixin): ) -> DataFrame: ... def filter( self, - items: list | None = ..., + items: ListLike | None = ..., like: _str | None = ..., regex: _str | None = ..., axis: Axis | None = ..., diff --git a/tests/test_frame.py b/tests/test_frame.py index 8ce0dc356..8d53da11e 100644 --- a/tests/test_frame.py +++ b/tests/test_frame.py @@ -319,6 +319,8 @@ def test_types_filter() -> None: df.filter(items=["col1"]) df.filter(regex="co.*") df.filter(like="1") + # [PR 964] Docs state `items` is `list-like` + df.filter(items=("col2", "col2", 1, tuple([4]))) def test_types_setting() -> None: