@@ -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