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 1a19b9801f..d16a547b82 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,9 @@ def _get_session(self, config: RepoConfig):
 
             # creation of Session
             self._keyspace = keyspace
-            self._session = self._cluster.connect(self._keyspace, wait_for_all_pools=True)
+            self._session = self._cluster.connect(
+                self._keyspace, wait_for_all_pools=True
+            )
 
         return self._session