spark rollfun add missing window size

duckdblabs · Jul 24, 2023 · 0da07f9 · 0da07f9
1 parent 088e89b
commit 0da07f9
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/spark/rollfun-spark.py b/spark/rollfun-spark.py
@@ -46,6 +46,11 @@
 
 x.createOrReplaceTempView("x")
 
+# window size
+w = int(x.count()/1e3)
+wsmall = int(x.count()/1e4)
+wbig = int(x.count()/1e2)
+
 task_init = timeit.default_timer()
 print("rolling...", flush=True)