fix: Waiting for all the connectiond to be eastablished at the beginning

ExpediaGroup · Jan 13, 2025 · d742474 · d742474
1 parent eb6660f
commit d742474
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/...python/feast/infra/online_stores/contrib/cassandra_online_store/cassandra_online_store.py b/...python/feast/infra/online_stores/contrib/cassandra_online_store/cassandra_online_store.py
@@ -416,7 +416,7 @@ def _get_session(self, config: RepoConfig):
 
             # creation of Session
             self._keyspace = keyspace
-            self._session = self._cluster.connect(self._keyspace)
+            self._session = self._cluster.connect(self._keyspace, wait_for_all_pools=True)
 
         return self._session