diff --git a/ibis_ml/steps/_select_features.py b/ibis_ml/steps/_select_features.py index a4b7919..d9407d0 100644 --- a/ibis_ml/steps/_select_features.py +++ b/ibis_ml/steps/_select_features.py @@ -67,7 +67,11 @@ def fit_table(self, table: ir.Table, metadata: Metadata) -> None: col_var = results[f"{col_name}_var"] if isinstance(table[col_name], ir.NumericColumn): # Check variance for numeric columns - if col_var is None or math.isnan(col_var) or col_var < self.tolerance: + if ( + col_var is None + or math.isnan(col_var) + or col_var < self.tolerance + ): cols.append(col_name) elif col_var is None or col_var < 2: # Check unique count for non-numeric columns