From 26b704a71f83ad0729f28e6eae05c7882f171701 Mon Sep 17 00:00:00 2001 From: Jiting Xu <126802425+jitingxu1@users.noreply.github.com> Date: Thu, 30 May 2024 06:56:29 -0700 Subject: [PATCH] fix(steps): set HandleUnivariateOutliers._fit_expr --- ibis_ml/steps/_handle_outliers.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/ibis_ml/steps/_handle_outliers.py b/ibis_ml/steps/_handle_outliers.py index dc625aa..acc767e 100644 --- a/ibis_ml/steps/_handle_outliers.py +++ b/ibis_ml/steps/_handle_outliers.py @@ -108,7 +108,8 @@ def fit_table(self, table: ir.Table, metadata: Metadata) -> None: aggs.append(c.quantile(0.25).name(f"{name}_25")) aggs.append(c.quantile(0.75).name(f"{name}_75")) - results = table.aggregate(aggs).execute().to_dict("records")[0] + self._fit_expr = [table.aggregate(aggs)] + results = self._fit_expr[0].execute().to_dict("records")[0] for name in columns: if self.method == "z-score":