From 4b33e0efc0cfb834fa1f59e89ae1a2de340c736e Mon Sep 17 00:00:00 2001 From: Erik Bernhardsson Date: Thu, 28 Jun 2018 13:24:48 -0400 Subject: [PATCH] forgot to push right code --- convoys/utils.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/convoys/utils.py b/convoys/utils.py index af35685..4e05ced 100644 --- 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: