forgot to push right code

better · Jun 28, 2018 · 4b33e0e · 4b33e0e
1 parent 47e03f9
commit 4b33e0e
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/convoys/utils.py b/convoys/utils.py
@@ -61,12 +61,12 @@ def get_arrays(data, features=None, groups=None, created=None,
 
     # First, construct either the `X` or the `G` array
     if features is None and groups is None:
-        if 'groups' in data.columns:
-            groups = 'groups'
+        if 'group' in data.columns:
+            groups = 'group'
         elif 'features' in data.columns:
             features = 'features'
         else:
-            raise Exception('Neither of the `features` or `groups` parameters'
+            raise Exception('Neither of the `features` or `group` parameters'
                             ' was provided, and there was no `features` or'
                             ' `groups` dataframe column')
     if groups is not None: