fix: Remove idle_heartbeat_interval from connection parameters and se…

…t it to zero in CassandraOnlineStore
ExpediaGroup · Jan 17, 2025 · bdfcf08 · bdfcf08
1 parent 141f218
commit bdfcf08
Showing 1 changed file with 2 additions 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
@@ -298,7 +298,6 @@ def _get_cluster(self, config: RepoConfig):
                 for k, v in {
                     "protocol_version": protocol_version,
                     "execution_profiles": execution_profiles,
-                    "idle_heartbeat_interval": None,
                 }.items()
                 if v is not None
             }
@@ -309,6 +308,8 @@ def _get_cluster(self, config: RepoConfig):
                     hosts,
                     port=port,
                     auth_provider=auth_provider,
+                    idle_heartbeat_interval=0,
+                    idle_heartbeat_timeout=0,
                     **cluster_kwargs,
                 )
             else: