small fix

ibis-project · Jan 25, 2024 · 5c40347 · 5c40347
1 parent e68b0db
commit 5c40347
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/ibis/backends/risingwave/tests/conftest.py b/ibis/backends/risingwave/tests/conftest.py
@@ -84,7 +84,10 @@ def connect(*, tmpdir, worker_id, port: int | None = None, **kw):
 
 @pytest.fixture(scope="session")
 def con(tmp_path_factory, data_dir, worker_id):
-    return TestConf.load_data(data_dir, tmp_path_factory, worker_id).connection
+    con = TestConf.load_data(data_dir, tmp_path_factory, worker_id).connection
+    cursor = con.raw_sql("SET RW_IMPLICIT_FLUSH TO true;")
+    cursor.close()
+    return con
 
 
 @pytest.fixture(scope="module")