From 9555ed429935fd0a70935a47be0b0b215b095a2f Mon Sep 17 00:00:00 2001 From: Huan Chen Date: Thu, 20 Mar 2025 21:33:28 +0000 Subject: [PATCH] perf: update shape to use quer_and_wait --- bigframes/session/executor.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/bigframes/session/executor.py b/bigframes/session/executor.py index e539525d80..1bea5cec02 100644 --- a/bigframes/session/executor.py +++ b/bigframes/session/executor.py @@ -462,7 +462,7 @@ def get_row_count(self, array_value: bigframes.core.ArrayValue) -> int: generate_row_count_plan(array_value.node) ) sql = self.compiler.compile(row_count_plan, ordered=False) - iter, _ = self._run_execute_query(sql) + iter, _ = self._run_execute_query(sql, query_with_job=False) return next(iter)[0] def cached(