From d742474d9dd07364a9265fc92e53198ce036e543 Mon Sep 17 00:00:00 2001
From: vbhagwat <vbhagwat@expediagroup.com>
Date: Mon, 13 Jan 2025 11:12:03 -0800
Subject: [PATCH] fix: Waiting for all the connectiond to be eastablished at
 the beginning

---
 .../contrib/cassandra_online_store/cassandra_online_store.py    | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/sdk/python/feast/infra/online_stores/contrib/cassandra_online_store/cassandra_online_store.py b/sdk/python/feast/infra/online_stores/contrib/cassandra_online_store/cassandra_online_store.py
index 9334bf49fe..1a19b9801f 100644
--- a/sdk/python/feast/infra/online_stores/contrib/cassandra_online_store/cassandra_online_store.py
+++ b/sdk/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