We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
2 parents ac56b79 + 791d4fb commit 92070d0Copy full SHA for 92070d0
1 file changed
merlin/systems/dag/ops/feast.py
@@ -275,6 +275,12 @@ def transform(self, df: InferenceDataFrame) -> InferenceDataFrame:
275
Transformed tensor dictionary
276
"""
277
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
+ )
284
entity_rows = [{self.entity_id: int(entity_id)} for entity_id in entity_ids]
285
286
feature_names = self.features + self.mh_features
0 commit comments