fix: Add null checks for session and cluster before shutdown in Cassa…

…ndraOnlineStore
ExpediaGroup · Jan 16, 2025 · 0c2e97c · 0c2e97c
1 parent 268f794
commit 0c2e97c
Showing 1 changed file with 6 additions and 4 deletions.
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
@@ -213,10 +213,12 @@ def shutdown(self):
         """
         Shutdown the Cassandra cluster and session.
         """
-        if not self._session.is_shutdown:
-            self._session.shutdown()
-        if not self._cluster.is_shutdown:
-            self._cluster.shutdown()
+        if self._session:
+            if not self._session.is_shutdown:
+                self._session.shutdown()
+        if self._cluster:
+            if not self._cluster.is_shutdown:
+                self._cluster.shutdown()
 
     def _get_cluster(self, config: RepoConfig):
         """