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":