From eb6660f65860f1b5f8ecadb9f06eca81c804061c Mon Sep 17 00:00:00 2001
From: Bhargav Dodla <bdodla@expediagroup.com>
Date: Mon, 13 Jan 2025 14:50:54 +0530
Subject: [PATCH] fix: Correctly register cluster shutdown on exit in
 CassandraOnlineStore

---
 .../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 899d8ff54f..9334bf49fe 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
@@ -311,7 +311,7 @@ def _get_cluster(self, config: RepoConfig):
             # creation of Session
             self._keyspace = keyspace
             # self._session = self._cluster.connect(self._keyspace)
-            atexit.register(self._cluster.shutdown())
+            atexit.register(self._cluster.shutdown)
 
         return self._cluster