lint

jitingxu1 · Jun 24, 2024 · 33bf160 · 33bf160
1 parent 3936fa2
commit 33bf160
Showing 1 changed file with 5 additions and 1 deletion.
diff --git 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