Skip to content

Commit 492c521

Browse files
authored
add data_source param to athena.repair_table (closes #1318) (#1324)
1 parent a2f68ef commit 492c521

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

awswrangler/athena/_utils.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -505,6 +505,7 @@ def start_query_execution(
505505
def repair_table(
506506
table: str,
507507
database: Optional[str] = None,
508+
data_source: Optional[str] = None,
508509
s3_output: Optional[str] = None,
509510
workgroup: Optional[str] = None,
510511
encryption: Optional[str] = None,
@@ -530,6 +531,8 @@ def repair_table(
530531
Table name.
531532
database : str, optional
532533
AWS Glue/Athena database name.
534+
data_source : Optional[str], optional
535+
Data Source / Catalog name. If None, 'AwsDataCatalog' is used.
533536
s3_output : str, optional
534537
AWS S3 path.
535538
workgroup : str, optional
@@ -559,6 +562,7 @@ def repair_table(
559562
query_id = start_query_execution(
560563
sql=query,
561564
database=database,
565+
data_source=data_source,
562566
s3_output=s3_output,
563567
workgroup=workgroup,
564568
encryption=encryption,

0 commit comments

Comments
 (0)