Merge pull request #210 from adkinsrs/master

Change to allow grouping order to be preserved
theislab · Nov 24, 2021 · 12f1286 · 12f1286
2 parents b8c6ae0 + 0464d2a
commit 12f1286
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/diffxpy/testing/utils.py b/diffxpy/testing/utils.py
@@ -111,7 +111,7 @@ def parse_grouping(data, sample_description, grouping):
 
 def split_x(data, grouping):
     grouping = np.asarray(grouping)
-    groups = np.unique(grouping)
+    groups = pd.Series(grouping).unique()
     x0 = data[np.where(grouping == groups[0])[0]]
     x1 = data[np.where(grouping == groups[1])[0]]
     return x0, x1