fix: add missing boto3 session in to_iceberg wait_query (#2428)

* fix: add missing boto3 session in to_iceberg wait_query --------- Signed-off-by: Abdel Jaidi <[email protected]>
aws · Aug 14, 2023 · ed14533 · ed14533
1 parent 8971415
commit ed14533
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/awswrangler/athena/_write_iceberg.py b/awswrangler/athena/_write_iceberg.py
@@ -54,7 +54,7 @@ def _create_iceberg_table(
         f"TBLPROPERTIES ('table_type' ='ICEBERG', 'format'='parquet'{table_properties_str})"
     )
 
-    query_id: str = _start_query_execution(
+    query_execution_id: str = _start_query_execution(
         sql=create_sql,
         workgroup=workgroup,
         wg_config=wg_config,
@@ -64,7 +64,7 @@ def _create_iceberg_table(
         kms_key=kms_key,
         boto3_session=boto3_session,
     )
-    wait_query(query_id)
+    wait_query(query_execution_id=query_execution_id, boto3_session=boto3_session)
 
 
 @apply_configs