Support q06

xorbitsai · Sep 19, 2023 · 692fb94 · 692fb94
1 parent 2c86ca7
commit 692fb94
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/xorbits_sql/executor.py b/xorbits_sql/executor.py
@@ -355,7 +355,7 @@ def aggregate(
             if step.group:
                 result = df.groupby(group_by).agg(**aggregations).reset_index()
             else:
-                result = df.agg(**aggregations)
+                result = df.agg(**aggregations).reset_index(drop=True)
             result.columns = names
         else:
             assert len(group_by) == len(names)

diff --git a/xorbits_sql/tests/test_tpc_h.py b/xorbits_sql/tests/test_tpc_h.py
@@ -54,7 +54,7 @@ def _to_csv(expression: exp.Expression) -> exp.Expression:
 
 def test_execute_tpc_h(prepare_data):
     conn, sqls = prepare_data
-    for sql, _ in sqls[:5]:
+    for sql, _ in sqls[:6]:
         expected = conn.execute(sql).fetchdf()
         result = execute(
             parse_one(sql, dialect="duckdb").transform(_to_csv).sql(pretty=True),