From 3a94013f0b63fecd1eec9e88c5554e9844a9fea9 Mon Sep 17 00:00:00 2001 From: Gyumin Lee Date: Fri, 23 Jun 2023 14:49:00 +0900 Subject: [PATCH] update Catalog argument in athena connector Signed-off-by: Gyumin Lee --- sdk/python/feast/infra/utils/aws_utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/sdk/python/feast/infra/utils/aws_utils.py b/sdk/python/feast/infra/utils/aws_utils.py index 5095ef42219..f48dfbb86b4 100644 --- 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, )