[TPCH] Polars turn off streaming (#1512)

Author: fjetter

tests/tpch/test_polars.py

@@ -58,7 +58,7 @@ def _():
                 ],
             )
             .sort(["l_returnflag", "l_linestatus"])
-        ).collect(streaming=True)
+        ).collect()
 
     run(_)
 
@@ -111,7 +111,7 @@ def _():
             )
             .limit(100)
             .with_columns(pl.col(pl.datatypes.Utf8).str.strip_chars().keep_name())
-        ).collect(streaming=True)
+        ).collect()
 
     run(_)
 
@@ -148,7 +148,7 @@ def _():
             )
             .sort(by=["revenue", "o_orderdate"], descending=[True, False])
             .limit(10)
-        ).collect(streaming=True)
+        ).collect()
 
     run(_)
 
@@ -170,7 +170,7 @@ def _():
             .agg(pl.len().alias("order_count"))
             .sort(by="o_orderpriority")
             .with_columns(pl.col("order_count").cast(pl.datatypes.Int64))
-        ).collect(streaming=True)
+        ).collect()
 
     run(_)
 
@@ -208,7 +208,7 @@ def _():
             .group_by("n_name")
             .agg([pl.sum("revenue")])
             .sort(by="revenue", descending=True)
-        ).collect(streaming=True)
+        ).collect()
 
     run(_)
 
@@ -231,7 +231,7 @@ def _():
                 (pl.col("l_extendedprice") * pl.col("l_discount")).alias("revenue")
             )
             .select(pl.sum("revenue").alias("revenue"))
-        ).collect(streaming=True)
+        ).collect()
 
     run(_)
 
@@ -280,7 +280,7 @@ def _():
             .group_by(["supp_nation", "cust_nation", "l_year"])
             .agg([pl.sum("volume").alias("revenue")])
             .sort(by=["supp_nation", "cust_nation", "l_year"])
-        ).collect(streaming=True)
+        ).collect()
 
     run(_)
 
@@ -332,7 +332,7 @@ def _():
             .agg((pl.sum("_tmp") / pl.sum("volume")).round(2).alias("mkt_share"))
             .sort("o_year")
         )
-        q_final.collect(streaming=True)
+        q_final.collect()
 
     run(_)
 
@@ -371,7 +371,7 @@ def _():
             .agg(pl.sum("amount").round(2).alias("sum_profit"))
             .sort(by=["nation", "o_year"], descending=[False, True])
         )
-        q_final.collect(streaming=True)
+        q_final.collect()
 
     run(_)
 
@@ -430,7 +430,7 @@ def _():
             .sort(by="revenue", descending=True)
             .limit(20)
         )
-        q_final.collect(streaming=True)
+        q_final.collect()
 
     run(_)
 
@@ -471,7 +471,7 @@ def _():
             .select(["ps_partkey", "value"])
             .sort("value", descending=True)
         )
-        q_final.collect(streaming=True)
+        q_final.collect()
 
     run(_)
 
@@ -510,7 +510,7 @@ def _():
             .agg([pl.col("high_line_count").sum(), pl.col("low_line_count").sum()])
             .sort("l_shipmode")
         )
-        q_final.collect(streaming=True)
+        q_final.collect()
 
     run(_)
 
@@ -535,7 +535,7 @@ def _():
             .select([pl.col("c_count"), pl.col("len").alias("custdist")])
             .sort(["custdist", "c_count"], descending=[True, True])
         )
-        q_final.collect(streaming=True)
+        q_final.collect()
 
     run(_)
 
@@ -564,7 +564,7 @@ def _():
                 .alias("promo_revenue")
             )
         )
-        q_final.collect(streaming=True)
+        q_final.collect()
 
     run(_)
 
@@ -597,7 +597,7 @@ def _():
             .select(["s_suppkey", "s_name", "s_address", "s_phone", "total_revenue"])
             .sort("s_suppkey")
         )
-        q_final.collect(streaming=True)
+        q_final.collect()
 
     run(_)
 
@@ -628,7 +628,7 @@ def _():
                 descending=[True, False, False, False],
             )
         )
-        q_final.collect(streaming=True)
+        q_final.collect()
 
     run(_)
 
@@ -657,7 +657,7 @@ def _():
                 (pl.col("l_extendedprice").sum() / 7.0).round(2).alias("avg_yearly")
             )
         )
-        q_final.collect(streaming=True)
+        q_final.collect()
 
     run(_)
 
@@ -693,7 +693,7 @@ def _():
             .sort(["o_totalprice", "o_orderdat"], descending=[True, False])
             .limit(100)
         )
-        q_final.collect(streaming=True)
+        q_final.collect()
 
     run(_)
 
@@ -740,7 +740,7 @@ def _():
                 .alias("revenue")
             )
         )
-        q_final.collect(streaming=True)
+        q_final.collect()
 
     run(_)
 
@@ -784,7 +784,7 @@ def _():
             .select(["s_name", "s_address"])
             .sort("s_name")
         )
-        q_final.collect(streaming=True)
+        q_final.collect()
 
     run(_)
 
@@ -823,7 +823,7 @@ def _():
             .sort(by=["numwait", "s_name"], descending=[True, False])
             .limit(100)
         )
-        q_final.collect(streaming=True)
+        q_final.collect()
 
     run(_)
 
@@ -866,6 +866,6 @@ def _():
             )
             .sort("cntrycode")
         )
-        q_final.collect(streaming=True)
+        q_final.collect()
 
     run(_)