Skip to content

Commit 75f3067

Browse files
Get feast repo path from feature store (NVIDIA-Merlin#84)
Co-authored-by: Karl Higley <kmhigley@gmail.com>
1 parent 6d2471e commit 75f3067

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

merlin/systems/dag/ops/feast.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -33,7 +33,6 @@ class QueryFeast(PipelineableInferenceOperator):
3333
def from_feature_view(
3434
cls,
3535
store: FeatureStore,
36-
path: str,
3736
view: str,
3837
column: str,
3938
output_prefix: str = None,
@@ -109,7 +108,7 @@ def from_feature_view(
109108
entity_id, dtype=entity_dtype, is_list=ent_is_list, is_ragged=ent_is_ragged
110109
)
111110
return QueryFeast(
112-
path,
111+
str(store.repo_path),
113112
entity_id,
114113
view,
115114
column,
@@ -185,6 +184,7 @@ def __init__(
185184
def compute_output_schema(
186185
self, input_schema: Schema, col_selector: ColumnSelector, prev_output_schema: Schema = None
187186
) -> Schema:
187+
"""Compute the output schema for the operator."""
188188
return self.output_schema
189189

190190
def compute_input_schema(
@@ -194,10 +194,12 @@ def compute_input_schema(
194194
deps_schema: Schema,
195195
selector: ColumnSelector,
196196
) -> Schema:
197+
"""Compute the input schema for the operator."""
197198
return self.input_schema
198199

199200
@classmethod
200201
def from_config(cls, config):
202+
"""Create the operator from a config."""
201203
parameters = json.loads(config.get("params", ""))
202204
entity_id = parameters["entity_id"]
203205
entity_view = parameters["entity_view"]

0 commit comments

Comments
 (0)