From fa0c4944eef78e3d06ae547929a5e88d8735a531 Mon Sep 17 00:00:00 2001 From: Tsotne Tabidze Date: Thu, 15 Jul 2021 11:52:47 -0700 Subject: [PATCH] Rename method Signed-off-by: Tsotne Tabidze --- sdk/python/feast/infra/offline_stores/redshift.py | 2 +- sdk/python/feast/infra/utils/aws_utils.py | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/sdk/python/feast/infra/offline_stores/redshift.py b/sdk/python/feast/infra/offline_stores/redshift.py index 23f3dc58cb..c97389ceaa 100644 --- a/sdk/python/feast/infra/offline_stores/redshift.py +++ b/sdk/python/feast/infra/offline_stores/redshift.py @@ -151,7 +151,7 @@ def to_arrow(self) -> pa.Table: def to_s3(self) -> str: """ Export dataset to S3 in Parquet format and return path """ - aws_utils.unload_redshift_query_to_s3( + aws_utils.execute_redshift_query_and_unload_to_s3( self._redshift_client, self._config.offline_store.cluster_id, self._config.offline_store.database, diff --git a/sdk/python/feast/infra/utils/aws_utils.py b/sdk/python/feast/infra/utils/aws_utils.py index c8202ed69a..235f427b76 100644 --- a/sdk/python/feast/infra/utils/aws_utils.py +++ b/sdk/python/feast/infra/utils/aws_utils.py @@ -228,7 +228,7 @@ def delete_s3_directory(s3_resource, bucket: str, key: str): obj.delete() -def unload_redshift_query_to_s3( +def execute_redshift_query_and_unload_to_s3( redshift_data_client, cluster_id: str, database: str, @@ -262,7 +262,7 @@ def unload_redshift_query_to_pa( """ Unload Redshift Query results to S3 and get the results in PyArrow Table format """ bucket, key = get_bucket_and_key(s3_path) - unload_redshift_query_to_s3( + execute_redshift_query_and_unload_to_s3( redshift_data_client, cluster_id, database, user, s3_path, iam_role, query )