Skip to content

Commit 613d02c

Browse files
committed
revert executor change
1 parent cb99ca1 commit 613d02c

File tree

1 file changed

+2
-8
lines changed

1 file changed

+2
-8
lines changed

bigframes/session/bq_caching_executor.py

Lines changed: 2 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -196,10 +196,7 @@ def execute(
196196
self._publisher.publish(bigframes.core.events.ExecutionStarted())
197197

198198
# TODO: Support export jobs in combination with semi executors
199-
if (
200-
execution_spec.destination_spec is None
201-
and execution_spec.promise_under_10gb
202-
):
199+
if execution_spec.destination_spec is None:
203200
plan = self.prepare_plan(array_value.node, target="simplify")
204201
for exec in self._semi_executors:
205202
maybe_result = exec.execute(
@@ -676,14 +673,11 @@ def _execute_plan_gbq(
676673
)
677674

678675
table_info: Optional[bigquery.Table] = None
679-
total_rows = None
680676
if query_job and query_job.destination:
681677
table_info = self.bqclient.get_table(query_job.destination)
682678
size_bytes = table_info.num_bytes
683-
total_rows = table_info.num_rows
684679
else:
685680
size_bytes = None
686-
total_rows = iterator.total_rows
687681

688682
# we could actually cache even when caching is not explicitly requested, but being conservative for now
689683
if cache_spec is not None:
@@ -710,7 +704,7 @@ def _execute_plan_gbq(
710704
schema=og_schema,
711705
query_job=query_job,
712706
total_bytes=size_bytes,
713-
total_rows=total_rows,
707+
total_rows=iterator.total_rows,
714708
total_bytes_processed=iterator.total_bytes_processed,
715709
)
716710

0 commit comments

Comments
 (0)