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

Tsotne Tabidze · web-flow · commit 74f887f72e10 · 2022-03-04T11:53:58.000-08:00
Signed-off-by: Tsotne Tabidze &lt;tsotne@tecton.ai&gt;
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:

Original file line number	Diff line number	Diff line change
`@@ -11,7 +11,6 @@`
`11`	`11`	`retry,`
`12`	`12`	`retry_if_exception_type,`
`13`	`13`	`stop_after_attempt,`
`14`		`- stop_after_delay,`
`15`	`14`	`wait_exponential,`
`16`	`15`	`)`
`17`	`16`
`@@ -103,7 +102,6 @@ class RedshiftStatementNotFinishedError(Exception):`
`103`	`102`	`@retry(`
`104`	`103`	`wait=wait_exponential(multiplier=1, max=30),`
`105`	`104`	`retry=retry_if_exception_type(RedshiftStatementNotFinishedError),`
`106`		`- stop=stop_after_delay(300), # 300 seconds`
`107`	`105`	`reraise=True,`
`108`	`106`	`)`
`109`	`107`	`def wait_for_redshift_statement(redshift_data_client, statement: dict) -> None:`