Skip to content
Prev Previous commit
Next Next commit
Fix
Signed-off-by: Kevin Zhang <kzhang@tecton.ai>
  • Loading branch information
kevjumba committed Apr 13, 2022
commit 8d62f0c88cfa6b1e6186ff96033717f96b84a4e5
6 changes: 3 additions & 3 deletions sdk/python/feast/on_demand_feature_view.py
Original file line number Diff line number Diff line change
Expand Up @@ -420,9 +420,9 @@ def infer_features(self):
)
df[f"{feature.name}"] = pd.Series(dtype=dtype)
for request_data in self.source_request_sources.values():
for feature_name, feature_type in request_data.schema.items():
dtype = feast_value_type_to_pandas_type(feature_type)
df[f"{feature_name}"] = pd.Series(dtype=dtype)
for field in request_data.schema:
dtype = feast_value_type_to_pandas_type(field.dtype.to_value_type())
df[f"{field.name}"] = pd.Series(dtype=dtype)
output_df: pd.DataFrame = self.udf.__call__(df)
inferred_features = []
for f, dt in zip(output_df.columns, output_df.dtypes):
Expand Down
5 changes: 4 additions & 1 deletion sdk/python/feast/types.py
Original file line number Diff line number Diff line change
Expand Up @@ -81,7 +81,10 @@ def __str__(self):
return PRIMITIVE_FEAST_TYPES_TO_STRING[self.name]

def __eq__(self, other):
return self.value == other.value
if isinstance(other, PrimitiveFeastType):
return self.value == other.value
else:
return False

def __hash__(self):
return hash((PRIMITIVE_FEAST_TYPES_TO_STRING[self.name]))
Expand Down