Fix Redshift bug that stops waiting on statements after 5 minutes

Signed-off-by: Tsotne Tabidze <[email protected]>
feast-dev · Mar 4, 2022 · 0adf3eb · 0adf3eb
1 parent ec7385c
commit 0adf3eb
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: