From 869ded57a146761bb4dfc521a331bf0e31647e01 Mon Sep 17 00:00:00 2001 From: Yusuke Nishioka Date: Fri, 21 Apr 2023 10:36:39 +0900 Subject: [PATCH 1/2] Cache Bigtable client Signed-off-by: Yusuke Nishioka --- sdk/python/feast/infra/online_stores/bigtable.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/sdk/python/feast/infra/online_stores/bigtable.py b/sdk/python/feast/infra/online_stores/bigtable.py index e08bc44bdb..2fdc4b34eb 100644 --- a/sdk/python/feast/infra/online_stores/bigtable.py +++ b/sdk/python/feast/infra/online_stores/bigtable.py @@ -335,4 +335,6 @@ def teardown( def _get_client( self, online_config: BigtableOnlineStoreConfig, admin: bool = False ): - return bigtable.Client(project=online_config.project_id, admin=admin) + if self._client is None: + self._client = bigtable.Client(project=online_config.project_id, admin=admin) + return self._client From 2366c4ca4017cdd50783cbf6edc76faa2294b918 Mon Sep 17 00:00:00 2001 From: Yusuke Nishioka Date: Fri, 21 Apr 2023 12:33:40 +0900 Subject: [PATCH 2/2] Run format Signed-off-by: Yusuke Nishioka --- sdk/python/feast/infra/online_stores/bigtable.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/sdk/python/feast/infra/online_stores/bigtable.py b/sdk/python/feast/infra/online_stores/bigtable.py index 2fdc4b34eb..30561d0840 100644 --- a/sdk/python/feast/infra/online_stores/bigtable.py +++ b/sdk/python/feast/infra/online_stores/bigtable.py @@ -336,5 +336,7 @@ def _get_client( self, online_config: BigtableOnlineStoreConfig, admin: bool = False ): if self._client is None: - self._client = bigtable.Client(project=online_config.project_id, admin=admin) + self._client = bigtable.Client( + project=online_config.project_id, admin=admin + ) return self._client