Skip to content

Commit 92070d0

Browse files
authored
Merge branch 'main' into hugectr-base
2 parents ac56b79 + 791d4fb commit 92070d0

1 file changed

Lines changed: 6 additions & 0 deletions

File tree

merlin/systems/dag/ops/feast.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -275,6 +275,12 @@ def transform(self, df: InferenceDataFrame) -> InferenceDataFrame:
275275
Transformed tensor dictionary
276276
"""
277277
entity_ids = df[self.entity_column]
278+
279+
if len(entity_ids) < 1:
280+
raise ValueError(
281+
"No entity ids provided when querying Feast. Must provide "
282+
"at least one id in order to fetch features."
283+
)
278284
entity_rows = [{self.entity_id: int(entity_id)} for entity_id in entity_ids]
279285

280286
feature_names = self.features + self.mh_features

0 commit comments

Comments
 (0)