fix #62

tompollard · May 23, 2018 · 04c879c · 04c879c
1 parent 1672811
commit 04c879c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tableone.py b/tableone.py
@@ -93,8 +93,8 @@ def __init__(self, data, columns=None, categorical=None, groupby=None,
             raise InputError('Columns not found in dataset: {}'.format(notfound))
 
         # check for duplicate columns
-        dups = data[columns].columns.get_duplicates()
-        if dups:
+        dups = data[columns].columns[data[columns].columns.duplicated()].unique()
+        if not dups.empty:
             raise InputError('Input contains duplicate columns: {}'.format(dups))
 
         # if categorical not specified, try to identify categorical