Cast columns to list for compatibility with numpy 1.26+

hackingmaterials · Mar 26, 2024 · 802a119 · 802a119
1 parent ea849c5
commit 802a119
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/matminer/featurizers/base.py b/matminer/featurizers/base.py
@@ -334,6 +334,8 @@ def featurize_dataframe(
         # Check names to avoid overwriting the current columns
         # ConversionFeaturizer have attribute called _overwrite_data which
         # determines whether an Error is thrown
+        if not isinstance(labels, list):
+            labels = labels.tolist()
         overwrite = getattr(self, "_overwrite_data", False)
         if not overwrite:
             for col in df.columns.values: