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]>
zerafachris · Jul 3, 2023 · f6d3caf · f6d3caf
1 parent d372a09
commit f6d3caf
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,
         )