From 5c5204c8b641b58bb54bc5ed14610f7f2bac3be0 Mon Sep 17 00:00:00 2001
From: ritchie46 <ritchie46@gmail.com>
Date: Thu, 25 Dec 2025 13:37:05 +0100
Subject: [PATCH] Use Polars streaming

---
 polars-dataframe/query.py | 2 ++
 polars/query.py           | 2 ++
 2 files changed, 4 insertions(+)

diff --git a/polars-dataframe/query.py b/polars-dataframe/query.py
index 2b0e9439e..83c775246 100755
--- a/polars-dataframe/query.py
+++ b/polars-dataframe/query.py
@@ -6,6 +6,8 @@
 import json
 import os
 
+pl.Config.set_engine_affinity("streaming")
+
 # 0: No., 1: SQL, 2: Polars
 queries = [
     ("Q0", "SELECT COUNT(*) FROM hits;", lambda x: x.select(pl.len()).collect().height),
diff --git a/polars/query.py b/polars/query.py
index 0e8221668..d077e9814 100755
--- a/polars/query.py
+++ b/polars/query.py
@@ -6,6 +6,8 @@
 import json
 import os
 
+pl.Config.set_engine_affinity("streaming")
+
 # 0: No., 1: SQL, 2: Polars
 queries = [
     ("Q0", "SELECT COUNT(*) FROM hits;", lambda x: x.select(pl.len()).collect().height),