🐛 Patch the 'Category encoder bug'.

scikit-learn-contrib · Sep 3, 2024 · 8ddc3dd · 8ddc3dd
1 parent 574f3db
commit 8ddc3dd
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/qolmat/analysis/holes_characterization.py b/qolmat/analysis/holes_characterization.py
@@ -185,6 +185,9 @@ def _encode_dataframe(self, df: pd.DataFrame) -> np.ndarray:
             The encoded DataFrame as a numpy ndarray, with numeric data concatenated
             with one-hot encoded categorical and boolean data.
         """
+        if not df.select_dtypes(include=["object", "bool"]).columns.to_list():
+            return df.to_numpy()
+
         if not self.encoder:
             self.encoder = OneHotEncoder(
                 cols=df.select_dtypes(include=["object", "bool"]).columns,