diff --git a/sdk/python/feast/infra/offline_stores/bigquery.py b/sdk/python/feast/infra/offline_stores/bigquery.py index bcfc5f08f44..681c7e94787 100644 --- a/sdk/python/feast/infra/offline_stores/bigquery.py +++ b/sdk/python/feast/infra/offline_stores/bigquery.py @@ -286,13 +286,13 @@ def _wait_until_done(job_id): job_id = bq_job.job_id _wait_until_done(job_id=job_id) + if bq_job.exception(): + raise bq_job.exception() + if not _is_done(job_id): client.cancel_job(job_id) raise BigQueryJobCancelled(job_id=job_id) - if bq_job.exception(): - raise bq_job.exception() - @dataclass(frozen=True) class FeatureViewQueryContext: