Fix KeyError in get_group (#109)

red-data-tools · May 22, 2024 · 3ff4c85 · 3ff4c85
1 parent f79af86
commit 3ff4c85
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/lib/charty/table_adapters/pandas_adapter.rb b/lib/charty/table_adapters/pandas_adapter.rb
@@ -208,7 +208,12 @@ def [](key)
                 when PyCall::Tuple
                   key
                 else
-                  PyCall::Tuple.new(*Array(key))
+                  key = Array(key)
+                  if key.length == 1 && key[0].is_a?(Integer)
+                    key[0]
+                  else
+                    PyCall::Tuple.new(*key)
+                  end
                 end
           Charty::Table.new(@groupby.get_group(key))
         end