diff --git a/sdk/python/feast/infra/utils/aws_utils.py b/sdk/python/feast/infra/utils/aws_utils.py index b25454ca6a..e7f628795d 100644 --- a/sdk/python/feast/infra/utils/aws_utils.py +++ b/sdk/python/feast/infra/utils/aws_utils.py @@ -11,7 +11,6 @@ retry, retry_if_exception_type, stop_after_attempt, - stop_after_delay, wait_exponential, ) @@ -103,7 +102,6 @@ class RedshiftStatementNotFinishedError(Exception): @retry( wait=wait_exponential(multiplier=1, max=30), retry=retry_if_exception_type(RedshiftStatementNotFinishedError), - stop=stop_after_delay(300), # 300 seconds reraise=True, ) def wait_for_redshift_statement(redshift_data_client, statement: dict) -> None: