Use keepdims with scipy mode to preserve indexing

hackingmaterials · Feb 7, 2024 · 17e42b5 · 17e42b5
1 parent cd51b26
commit 17e42b5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/matminer/featurizers/utils/stats.py b/matminer/featurizers/utils/stats.py
@@ -242,7 +242,7 @@ def mode(data_lst, weights=None):
             mode
         """
         if weights is None:
-            return scipy.stats.mode(data_lst).mode[0]
+            return scipy.stats.mode(data_lst, keepdims=True).mode[0]
         else:
             # Find the entry(s) with the largest weight
             data_lst = np.array(data_lst)