code: def await_bucket(bucket_name):
      # Avoid a race condition after creating the S3 Bucket.
      client = boto3.client("s3")
      waiter = client.get_waiter("bucket_exists")
      waiter.wait(Bucket=bucket_name)
  
file_path: airflow/tests/system/providers/amazon/aws/example_athena.py
imports: from __future__ import annotations
  from datetime import datetime
  import boto3
  from airflow.decorators import task
  from airflow.models.baseoperator import chain
  from airflow.models.dag import DAG
  from airflow.providers.amazon.aws.hooks.s3 import S3Hook
  from airflow.providers.amazon.aws.operators.athena import AthenaOperator
  from airflow.providers.amazon.aws.operators.s3 import S3CreateBucketOperator, S3CreateObjectOperator, S3DeleteBucketOperator
  from airflow.providers.amazon.aws.sensors.athena import AthenaSensor
  from airflow.utils.trigger_rule import TriggerRule
  from tests.system.providers.amazon.aws.utils import SystemTestContextBuilder
  from tests.system.utils import get_test_run
label_desc: Use image node_id 1 for calling a global function with example usage: await_bucket(bucket_name) without return types