Skip to content

Commit

Permalink
Merge branch 'main' into hugectr-base
Browse files Browse the repository at this point in the history
  • Loading branch information
karlhigley authored Jul 15, 2022
2 parents ac56b79 + 791d4fb commit 92070d0
Showing 1 changed file with 6 additions and 0 deletions.
6 changes: 6 additions & 0 deletions merlin/systems/dag/ops/feast.py
Original file line number Diff line number Diff line change
Expand Up @@ -275,6 +275,12 @@ def transform(self, df: InferenceDataFrame) -> InferenceDataFrame:
Transformed tensor dictionary
"""
entity_ids = df[self.entity_column]

if len(entity_ids) < 1:
raise ValueError(
"No entity ids provided when querying Feast. Must provide "
"at least one id in order to fetch features."
)
entity_rows = [{self.entity_id: int(entity_id)} for entity_id in entity_ids]

feature_names = self.features + self.mh_features
Expand Down

0 comments on commit 92070d0

Please sign in to comment.