fix: Missing Catalog argument in athena connector (feast-dev#3661)

GyuminJack · Gyumin Lee · web-flow · commit f6d3caf8affc · 2023-07-03T12:32:10.000-07:00
update Catalog argument in athena connector

Signed-off-by: Gyumin Lee &lt;t1100394@T1100394PM01.local&gt;
Co-authored-by: Gyumin Lee &lt;t1100394@T1100394PM01.local&gt;
diff --git a/sdk/python/feast/infra/utils/aws_utils.py b/sdk/python/feast/infra/utils/aws_utils.py
@@ -755,7 +755,7 @@ def execute_athena_query_async(
         # return athena_data_client.execute_statement(
         return athena_data_client.start_query_execution(
             QueryString=query,
-            QueryExecutionContext={"Database": database},
+            QueryExecutionContext={"Database": database, "Catalog": data_source},
             WorkGroup=workgroup,
         )
 

Original file line number	Diff line number	Diff line change
`@@ -755,7 +755,7 @@ def execute_athena_query_async(`
`755`	`755`	`# return athena_data_client.execute_statement(`
`756`	`756`	`return athena_data_client.start_query_execution(`
`757`	`757`	`QueryString=query,`
`758`		`- QueryExecutionContext={"Database": database},`
	`758`	`+ QueryExecutionContext={"Database": database, "Catalog": data_source},`
`759`	`759`	`WorkGroup=workgroup,`
`760`	`760`	`)`
`761`	`761`