From e76be4177cd8cfb154bef86d561c0c69ca5052a1 Mon Sep 17 00:00:00 2001 From: Will Ayd Date: Sat, 17 Aug 2024 12:57:42 -0400 Subject: [PATCH] More narwhals --- tests/conftest.py | 10 ++-------- 1 file changed, 2 insertions(+), 8 deletions(-) diff --git a/tests/conftest.py b/tests/conftest.py index c0a6b026..a61b01d1 100644 --- a/tests/conftest.py +++ b/tests/conftest.py @@ -484,15 +484,9 @@ def drop_columns(frame, columns): return frame.drop(columns) @staticmethod + @nw.narwhalify def select_columns(frame, columns): - if isinstance(frame, pd.DataFrame): - return frame[columns] - elif isinstance(frame, pa.Table): - return frame.select(columns) - elif isinstance(frame, pl.DataFrame): - return frame.select(columns) - else: - raise NotImplementedError("select_columns not implemented for type") + frame.select(columns) @staticmethod @nw.narwhalify