Don't print removed cols when none are removed

carpentries-incubator · Aug 14, 2023 · 8dbc669 · 8dbc669
1 parent af4abec
commit 8dbc669
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/code/preprocessing.py b/code/preprocessing.py
@@ -100,7 +100,8 @@ def remove_bad_cols(X: Union[pd.Series, pd.DataFrame], limited_var_thresh: float
     final_cols = without_sparse.columns
     dropped_cols = orig_cols[~orig_cols.isin(final_cols)]
     print(f'{len(dropped_cols)} removed, {len(final_cols)} remaining.')
-    print(f'Columns removed: {list(dropped_cols)}')
+    if len(dropped_cols) > 0:
+        print(f'Columns removed: {list(dropped_cols)}')
 
     # Return
     return without_sparse