fix(steps): set HandleUnivariateOutliers._fit_expr

jitingxu1 · May 30, 2024 · c5d92fc · c5d92fc
1 parent 54eecde
commit c5d92fc
Showing 1 changed file with 2 additions and 1 deletion.
diff --git 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":