Update ibis_ml/steps/_select_features.py

Co-authored-by: Deepyaman Datta <[email protected]>
jitingxu1 · Jun 25, 2024 · 2f121d3 · 2f121d3
1 parent 5b1d7ba
commit 2f121d3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ibis_ml/steps/_select_features.py b/ibis_ml/steps/_select_features.py
@@ -69,7 +69,7 @@ def fit_table(self, table: ir.Table, metadata: Metadata) -> None:
                     # Check variance for numeric columns
                     if math.isnan(col_var) or col_var < self.tolerance:
                         cols.append(col_name)
-                elif col_var is None or col_var < 2:
+                elif col_var < 2:
                     # Check unique count for non-numeric columns
                     cols.append(col_name)
                     metadata.drop_categories(col_name)