From c74d4f353a516e524acec8d336b48e87b0d9f3bb Mon Sep 17 00:00:00 2001 From: Bhargav Dodla Date: Tue, 4 Feb 2025 16:10:34 -0800 Subject: [PATCH] refactor: Removed write limit --- .../contrib/cassandra_online_store/cassandra_online_store.py | 3 --- 1 file changed, 3 deletions(-) diff --git a/sdk/python/feast/infra/online_stores/contrib/cassandra_online_store/cassandra_online_store.py b/sdk/python/feast/infra/online_stores/contrib/cassandra_online_store/cassandra_online_store.py index 4b74a491361..43cff71cc65 100644 --- a/sdk/python/feast/infra/online_stores/contrib/cassandra_online_store/cassandra_online_store.py +++ b/sdk/python/feast/infra/online_stores/contrib/cassandra_online_store/cassandra_online_store.py @@ -19,7 +19,6 @@ """ import logging -import time from datetime import datetime from typing import Any, Callable, Dict, List, Literal, Optional, Sequence, Tuple @@ -353,7 +352,6 @@ def online_write_batch( display progress. """ write_concurrency = config.online_store.write_concurrency - write_limit = 1.0 / write_concurrency project = config.project ttl = ( table.online_store_key_ttl_seconds @@ -388,7 +386,6 @@ def online_write_batch( ) batch.add(insert_cql, params) futures.append(session.execute_async(batch)) - time.sleep(write_limit) # TODO: Make this efficient by leveraging continuous writes rather # than blocking until all writes are done. We may need to rate limit