diff --git a/sdk/python/feast/infra/offline_stores/bigquery.py b/sdk/python/feast/infra/offline_stores/bigquery.py index 55dc2d9cd4..e674786864 100644 --- a/sdk/python/feast/infra/offline_stores/bigquery.py +++ b/sdk/python/feast/infra/offline_stores/bigquery.py @@ -288,13 +288,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: