diff --git a/sdk/python/feast/infra/offline_stores/bigquery.py b/sdk/python/feast/infra/offline_stores/bigquery.py index fdeb28ef46..ffff021c09 100644 --- 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)