Skip to content

Commit 1a442cd

Browse files
committed
make sure start and end dates are required, otherwise it will not work
1 parent 50d84d1 commit 1a442cd

File tree

1 file changed

+7
-2
lines changed
  • sdk/python/feast/infra/offline_stores/contrib/clickhouse_offline_store

1 file changed

+7
-2
lines changed

sdk/python/feast/infra/offline_stores/contrib/clickhouse_offline_store/clickhouse.py

Lines changed: 7 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -201,13 +201,18 @@ def pull_all_from_table_or_query(
201201
join_key_columns: List[str],
202202
feature_name_columns: List[str],
203203
timestamp_field: str,
204-
start_date: datetime,
205-
end_date: datetime,
206204
created_timestamp_column: Optional[str] = None,
205+
start_date: Optional[datetime] = None,
206+
end_date: Optional[datetime] = None,
207207
) -> RetrievalJob:
208208
assert isinstance(config.offline_store, ClickhouseOfflineStoreConfig)
209209
assert isinstance(data_source, ClickhouseSource)
210210

211+
if start_date is None or end_date is None:
212+
raise ValueError(
213+
"Start_date and end_date must be provided. Pulling without filtering is not supported in ClickHouseOfflineStore."
214+
)
215+
211216
# https://github.com/feast-dev/feast/issues/5707
212217
# Not an ideal solution, but least invasion into existing codebase
213218
if config.offline_store.deduplicate_pushdown:

0 commit comments

Comments
 (0)