fix: Update RedisCluster to use redis-py official implementation (#2554)

* switch to using redis cluster in redis-py Signed-off-by: Kevin Zhang <[email protected]> * fix Signed-off-by: Kevin Zhang <[email protected]> * Fix Signed-off-by: Kevin Zhang <[email protected]> * Fix dependencies Signed-off-by: Kevin Zhang <[email protected]> * Update py3.7-ci-requirements.txt Signed-off-by: Felix Wang <[email protected]> * update 3.10 requirements Signed-off-by: Kevin Zhang <[email protected]> * Fix dependencies to address issues Signed-off-by: Kevin Zhang <[email protected]> * Fix Signed-off-by: Kevin Zhang <[email protected]> * Fix Signed-off-by: Kevin Zhang <[email protected]> Co-authored-by: Felix Wang <[email protected]>
feast-dev · Apr 20, 2022 · c47fa2a · c47fa2a
1 parent 5539c51
commit c47fa2a
Show file tree

Hide file tree

Showing 9 changed files with 425 additions and 685 deletions.
diff --git a/sdk/python/feast/infra/online_stores/redis.py b/sdk/python/feast/infra/online_stores/redis.py
@@ -42,7 +42,7 @@
 
 try:
     from redis import Redis
-    from rediscluster import RedisCluster
+    from redis.cluster import ClusterNode, RedisCluster
 except ImportError as e:
     from feast.errors import FeastExtrasDependencyImportError
 
@@ -164,7 +164,9 @@ def _get_client(self, online_store_config: RedisOnlineStoreConfig):
                 online_store_config.connection_string
             )
             if online_store_config.redis_type == RedisType.redis_cluster:
-                kwargs["startup_nodes"] = startup_nodes
+                kwargs["startup_nodes"] = [
+                    ClusterNode(**node) for node in startup_nodes
+                ]
                 self._client = RedisCluster(**kwargs)
             else:
                 kwargs["host"] = startup_nodes[0]["host"]