From c3a3addc9db8136ac7d37d4bbf1fea476da16156 Mon Sep 17 00:00:00 2001 From: Tsotne Tabidze Date: Fri, 4 Mar 2022 11:53:58 -0800 Subject: [PATCH] fix: Fix Redshift bug that stops waiting on statements after 5 minutes (#2363) Signed-off-by: Tsotne Tabidze Signed-off-by: Achal Shah --- sdk/python/feast/infra/utils/aws_utils.py | 2 -- 1 file changed, 2 deletions(-) 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: