From 73193afe57bec4e5b04a562eb1fed67d96ee67f5 Mon Sep 17 00:00:00 2001 From: Chester Ong Date: Thu, 8 Feb 2024 23:59:18 +0800 Subject: [PATCH] fix circular import Signed-off-by: Chester Ong --- sdk/python/feast/infra/utils/snowflake/snowflake_utils.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/sdk/python/feast/infra/utils/snowflake/snowflake_utils.py b/sdk/python/feast/infra/utils/snowflake/snowflake_utils.py index 193bd0d494..8eb5177ac2 100644 --- a/sdk/python/feast/infra/utils/snowflake/snowflake_utils.py +++ b/sdk/python/feast/infra/utils/snowflake/snowflake_utils.py @@ -22,8 +22,6 @@ import feast from feast.errors import SnowflakeIncompleteConfig, SnowflakeQueryUnknownError from feast.feature_view import FeatureView -from feast.infra.offline_stores.snowflake import SnowflakeOfflineStoreConfig -from feast.infra.online_stores.snowflake import SnowflakeOnlineStoreConfig from feast.repo_config import RepoConfig try: @@ -47,7 +45,7 @@ class GetSnowflakeConnection: def __init__( self, - config: SnowflakeOfflineStoreConfig | SnowflakeOnlineStoreConfig, + config: str, autocommit=True, ): self.config = config