File tree Expand file tree Collapse file tree 1 file changed +4
-2
lines changed Expand file tree Collapse file tree 1 file changed +4
-2
lines changed Original file line number Diff line number Diff line change @@ -33,7 +33,6 @@ class QueryFeast(PipelineableInferenceOperator):
33
33
def from_feature_view (
34
34
cls ,
35
35
store : FeatureStore ,
36
- path : str ,
37
36
view : str ,
38
37
column : str ,
39
38
output_prefix : str = None ,
@@ -109,7 +108,7 @@ def from_feature_view(
109
108
entity_id , dtype = entity_dtype , is_list = ent_is_list , is_ragged = ent_is_ragged
110
109
)
111
110
return QueryFeast (
112
- path ,
111
+ str ( store . repo_path ) ,
113
112
entity_id ,
114
113
view ,
115
114
column ,
@@ -185,6 +184,7 @@ def __init__(
185
184
def compute_output_schema (
186
185
self , input_schema : Schema , col_selector : ColumnSelector , prev_output_schema : Schema = None
187
186
) -> Schema :
187
+ """Compute the output schema for the operator."""
188
188
return self .output_schema
189
189
190
190
def compute_input_schema (
@@ -194,10 +194,12 @@ def compute_input_schema(
194
194
deps_schema : Schema ,
195
195
selector : ColumnSelector ,
196
196
) -> Schema :
197
+ """Compute the input schema for the operator."""
197
198
return self .input_schema
198
199
199
200
@classmethod
200
201
def from_config (cls , config ):
202
+ """Create the operator from a config."""
201
203
parameters = json .loads (config .get ("params" , "" ))
202
204
entity_id = parameters ["entity_id" ]
203
205
entity_view = parameters ["entity_view" ]
You can’t perform that action at this time.
0 commit comments