File tree Expand file tree Collapse file tree 1 file changed +2
-8
lines changed
Expand file tree Collapse file tree 1 file changed +2
-8
lines changed Original file line number Diff line number Diff 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
You can’t perform that action at this time.
0 commit comments