Fix for FutrueWarning of get_group

red-data-tools · May 17, 2024 · f6009ce · f6009ce
1 parent 8b6c466
commit f6009ce
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/lib/charty/table_adapters/pandas_adapter.rb b/lib/charty/table_adapters/pandas_adapter.rb
@@ -204,6 +204,12 @@ def apply(*args, &block)
         end
 
         def [](key)
+          key = case key
+                when PyCall::Tuple
+                  key
+                else
+                  PyCall::Tuple.new(*key.to_a)
+                end
           Charty::Table.new(@groupby.get_group(key))
         end
       end