From 33bf1603bd07676ce17226beb98e56bfa442cddc Mon Sep 17 00:00:00 2001
From: Jiting Xu <126802425+jitingxu1@users.noreply.github.com>
Date: Sun, 23 Jun 2024 21:22:22 -0700
Subject: [PATCH] lint

---
 ibis_ml/steps/_select_features.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

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