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

update Catalog argument in athena connector Signed-off-by: Gyumin Lee <[email protected]> Co-authored-by: Gyumin Lee <[email protected]>
ExpediaGroup · Jun 11, 2024 · 0af9130 · 0af9130
1 parent be3281e
commit 0af9130
Showing 1 changed file with 1 addition and 1 deletion.
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,
         )