Fix dry_run bug that was making to_bigquery hang indefinitely (feast-…

…dev#1706) Signed-off-by: Cody Lin <[email protected]> Signed-off-by: CS <[email protected]>
8bit-pixies · Jul 16, 2021 · 697f054 · 697f054
1 parent c047186
commit 697f054
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/sdk/python/feast/infra/offline_stores/bigquery.py b/sdk/python/feast/infra/offline_stores/bigquery.py
@@ -282,17 +282,17 @@ def to_bigquery(self, job_config: bigquery.QueryJobConfig = None) -> Optional[st
 
         bq_job = self.client.query(self.query, job_config=job_config)
 
-        block_until_done(client=self.client, bq_job=bq_job)
-
-        if bq_job.exception():
-            raise bq_job.exception()
-
         if job_config.dry_run:
             print(
                 "This query will process {} bytes.".format(bq_job.total_bytes_processed)
             )
             return None
 
+        block_until_done(client=self.client, bq_job=bq_job)
+
+        if bq_job.exception():
+            raise bq_job.exception()
+
         print(f"Done writing to '{job_config.destination}'.")
         return str(job_config.destination)