fix: Disable idle heartbeat and shard awareness in CassandraOnlineStore

ExpediaGroup · Jan 10, 2025 · 26a4b70 · 26a4b70
1 parent 3580383
commit 26a4b70
Showing 1 changed file with 1 addition 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,7 +286,7 @@ def _get_session(self, config: RepoConfig):
                 for k, v in {
                     "protocol_version": protocol_version,
                     "execution_profiles": execution_profiles,
-                    "idle_heartbeat_interval": 5,
+                    "idle_heartbeat_interval": 0,
                 }.items()
                 if v is not None
             }
@@ -301,7 +301,6 @@ def _get_session(self, config: RepoConfig):
                 self._cluster = Cluster(
                     cloud={"secure_connect_bundle": secure_bundle_path},
                     auth_provider=auth_provider,
-                    shard_aware_options=dict(disable=True),
                     **cluster_kwargs,
                 )