Skip to content

Commit

Permalink
[AWS hook] use provided client to get the official waiter on fallback (
Browse files Browse the repository at this point in the history
…#31748)

we use the provided client to create custom waiters, but we need to use it for "official" ones too
Otherwise, passing an async client here still returns a synchronous waiter.
  • Loading branch information
vandonr-amz committed Jun 7, 2023
1 parent 0261bc5 commit a3768b4
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion airflow/providers/amazon/aws/hooks/base_aws.py
Expand Up @@ -870,7 +870,7 @@ def get_waiter(
)
# If there is no custom waiter found for the provided name,
# then try checking the service's official waiters.
return self.conn.get_waiter(waiter_name)
return client.get_waiter(waiter_name)

@staticmethod
def _apply_parameters_value(config: dict, waiter_name: str, parameters: dict[str, str] | None) -> dict:
Expand Down

0 comments on commit a3768b4

Please sign in to comment.