feat(amazon): change how poll_interval is loaded in redshift cluster …

…sensor
astronomer · Jan 25, 2024 · ae03f35 · ae03f35
1 parent 3c9313c
commit ae03f35
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/astronomer/providers/amazon/aws/sensors/redshift_cluster.py b/astronomer/providers/amazon/aws/sensors/redshift_cluster.py
@@ -13,13 +13,11 @@ class RedshiftClusterSensorAsync(RedshiftClusterSensor):
 
     def __init__(
         self,
-        *,
-        poll_interval: float = 5,
         **kwargs: Any,
     ):
         # TODO: Remove once deprecated
-        if poll_interval:
-            kwargs["poke_interval"] = poll_interval
+        if kwargs.get("poll_interval"):
+            kwargs["poke_interval"] = kwargs["poll_interval"]
             warnings.warn(
                 "Argument `poll_interval` is deprecated and will be removed "
                 "in a future release.  Please use  `poke_interval` instead.",