From d962faf93727a72e0542b2439adf4de5a2c24b0f Mon Sep 17 00:00:00 2001 From: Gil Forsyth Date: Mon, 6 Feb 2023 14:43:27 -0500 Subject: [PATCH] fix(polars): polars execute respect limit kwargs --- ibis/backends/polars/__init__.py | 11 +++++++++-- 1 file changed, 9 insertions(+), 2 deletions(-) diff --git a/ibis/backends/polars/__init__.py b/ibis/backends/polars/__init__.py index 9a72e3c39146..1740d40e34c4 100644 --- a/ibis/backends/polars/__init__.py +++ b/ibis/backends/polars/__init__.py @@ -273,10 +273,17 @@ def execute( self, expr: ir.Expr, params: Mapping[ir.Expr, object] = None, - limit: str = 'default', + limit: int | None = None, **kwargs: Any, ): - df = self.compile(expr, params=params).collect() + lf = self.compile(expr, params=params, **kwargs) + if limit == "default": + limit = ibis.options.sql.default_limit + if limit is not None: + df = lf.fetch(limit) + else: + df = lf.collect() + if isinstance(expr, ir.Table): return df.to_pandas() elif isinstance(expr, ir.Column):