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 (
@@ -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
You can’t perform that action at this time.
0 commit comments