fix: Fix Redshift bug that stops waiting on statements after 5 minutes (

feast-dev#2363) Signed-off-by: Tsotne Tabidze <tsotne@tecton.ai> Signed-off-by: Achal Shah <achals@gmail.com>
achals · Mar 8, 2022 · 5e5e10b · 5e5e10b
1 parent 7a9f186
commit 5e5e10b
Showing 1 changed file with 0 additions and 2 deletions.
diff --git 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: