fix: Set idle heartbeat interval to None in CassandraOnlineStore conf…

…iguration
ExpediaGroup · Jan 10, 2025 · c810ee8 · c810ee8
1 parent 26a4b70
commit c810ee8
Showing 1 changed file with 5 additions and 2 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
@@ -286,15 +286,18 @@ def _get_session(self, config: RepoConfig):
                 for k, v in {
                     "protocol_version": protocol_version,
                     "execution_profiles": execution_profiles,
-                    "idle_heartbeat_interval": 0,
+                    "idle_heartbeat_interval": None,
                 }.items()
                 if v is not None
             }
 
             # creation of Cluster (Cassandra vs. Astra)
             if hosts:
                 self._cluster = Cluster(
-                    hosts, port=port, auth_provider=auth_provider, **cluster_kwargs
+                    hosts,
+                    port=port,
+                    auth_provider=auth_provider,
+                    **cluster_kwargs,
                 )
             else:
                 # we use 'secure_bundle_path'