Update group to store a list, preventing pandas from throwing FutureW…

…arnings on group_by
kieferk · Aug 17, 2018 · c606c5c · c606c5c
1 parent f80460e
commit c606c5c
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/dfply/group.py b/dfply/group.py
@@ -4,7 +4,7 @@
 @pipe
 @symbolic_evaluation(eval_as_label=True)
 def group_by(df, *args):
-    df._grouped_by = args
+    df._grouped_by = list(args)
     return df
 
 

diff --git a/test/test_group.py b/test/test_group.py
@@ -10,4 +10,4 @@
 def test_group_attributes():
     d = diamonds >> group_by('cut')
     assert hasattr(d, '_grouped_by')
-    assert d._grouped_by == ('cut',)
+    assert d._grouped_by == ['cut',]