diff --git a/sdk/python/feast/infra/offline_stores/bigquery.py b/sdk/python/feast/infra/offline_stores/bigquery.py index 8cdcb39694..4ca8f390b4 100644 --- a/sdk/python/feast/infra/offline_stores/bigquery.py +++ b/sdk/python/feast/infra/offline_stores/bigquery.py @@ -207,7 +207,7 @@ def block_until_done( client: Client, bq_job: Union[bigquery.job.query.QueryJob, bigquery.job.load.LoadJob], timeout: int = 1800, - retry_cadence: int = 10, + retry_cadence: float = 1, ): """ Waits for bq_job to finish running, up to a maximum amount of time specified by the timeout parameter (defaulting to 30 minutes). @@ -226,7 +226,7 @@ def block_until_done( # For test environments, retry more aggressively is_test = os.getenv("IS_TEST", default="False") == "True" if is_test: - retry_cadence = 1 + retry_cadence = 0.1 def _wait_until_done(job_id): if client.get_job(job_id).state in ["PENDING", "RUNNING"]: