fix: Correctly register cluster shutdown on exit in CassandraOnlineStore

ExpediaGroup · Jan 13, 2025 · eb6660f · eb6660f
1 parent 482b236
commit eb6660f
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
@@ -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