From 6eea85300a11946b5dac49019bd5190770f2f398 Mon Sep 17 00:00:00 2001 From: Jim Crist-Harif Date: Mon, 17 Jul 2023 10:12:59 -0500 Subject: [PATCH] fix: forward arguments through `__dataframe__` protocol --- ibis/expr/types/relations.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/ibis/expr/types/relations.py b/ibis/expr/types/relations.py index 9909466ade9c..d7b67134efeb 100644 --- a/ibis/expr/types/relations.py +++ b/ibis/expr/types/relations.py @@ -109,8 +109,8 @@ class Table(Expr, _FixedTextJupyterMixin): def __array__(self, dtype=None): return self.execute().__array__(dtype) - def __dataframe__(self): - return self.to_pyarrow().__dataframe__() + def __dataframe__(self, *args: Any, **kwargs: Any): + return self.to_pyarrow().__dataframe__(*args, **kwargs) def as_table(self) -> Table: """Promote the expression to a table.