Skip to content

Commit bde45e4

Browse files
committed
revert executor change
1 parent 5d5e697 commit bde45e4

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(
@@ -680,14 +677,11 @@ def _execute_plan_gbq(
680677
)
681678

682679
table_info: Optional[bigquery.Table] = None
683-
total_rows = None
684680
if query_job and query_job.destination:
685681
table_info = self.bqclient.get_table(query_job.destination)
686682
size_bytes = table_info.num_bytes
687-
total_rows = table_info.num_rows
688683
else:
689684
size_bytes = None
690-
total_rows = iterator.total_rows
691685

692686
# we could actually cache even when caching is not explicitly requested, but being conservative for now
693687
if cache_spec is not None:
@@ -714,7 +708,7 @@ def _execute_plan_gbq(
714708
schema=og_schema,
715709
query_job=query_job,
716710
total_bytes=size_bytes,
717-
total_rows=total_rows,
711+
total_rows=iterator.total_rows,
718712
total_bytes_processed=iterator.total_bytes_processed,
719713
)
720714

0 commit comments

Comments
 (0)