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