fix: forward arguments through __dataframe__ protocol

ibis-project · Jul 17, 2023 · 50f3be9 · 50f3be9
1 parent 603cb6a
commit 50f3be9
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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.