Skip to content

Commit bb6b34c

Browse files
committed
revert executor change
1 parent 0525054 commit bb6b34c

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
@@ -189,10 +189,7 @@ def execute(
189189
execution_spec: ex_spec.ExecutionSpec,
190190
) -> executor.ExecuteResult:
191191
# TODO: Support export jobs in combination with semi executors
192-
if (
193-
execution_spec.destination_spec is None
194-
and execution_spec.promise_under_10gb
195-
):
192+
if execution_spec.destination_spec is None:
196193
plan = self.prepare_plan(array_value.node, target="simplify")
197194
for exec in self._semi_executors:
198195
maybe_result = exec.execute(
@@ -650,14 +647,11 @@ def _execute_plan_gbq(
650647
)
651648

652649
table_info: Optional[bigquery.Table] = None
653-
total_rows = None
654650
if query_job and query_job.destination:
655651
table_info = self.bqclient.get_table(query_job.destination)
656652
size_bytes = table_info.num_bytes
657-
total_rows = table_info.num_rows
658653
else:
659654
size_bytes = None
660-
total_rows = iterator.total_rows
661655

662656
# we could actually cache even when caching is not explicitly requested, but being conservative for now
663657
if cache_spec is not None:
@@ -684,7 +678,7 @@ def _execute_plan_gbq(
684678
schema=og_schema,
685679
query_job=query_job,
686680
total_bytes=size_bytes,
687-
total_rows=total_rows,
681+
total_rows=iterator.total_rows,
688682
total_bytes_processed=iterator.total_bytes_processed,
689683
)
690684

0 commit comments

Comments
 (0)