Skip to content
Prev Previous commit
Next Next commit
Fix
Signed-off-by: Kevin Zhang <kzhang@tecton.ai>
  • Loading branch information
kevjumba committed Apr 21, 2022
commit e0078b125a004930c819f20e51e3e1b84bbfb435
6 changes: 2 additions & 4 deletions sdk/python/feast/on_demand_feature_view.py
Original file line number Diff line number Diff line change
Expand Up @@ -244,13 +244,11 @@ def proto_class(self) -> Type[OnDemandFeatureViewProto]:
return OnDemandFeatureViewProto

def __copy__(self):

fv = OnDemandFeatureView(
name=self.name,
schema=self.features,
sources=list(
**self.source_feature_view_projections.values(),
**self.source_request_sources.values(),
),
sources=list(self.source_feature_view_projections.values()) + list(self.source_request_sources.values()),
udf=self.udf,
description=self.description,
tags=self.tags,
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -151,7 +151,7 @@ def create_item_embeddings_feature_view(source, infer_features: bool = False):
return item_embeddings_feature_view


def create_item_embeddings_batch_feature_view(source, infer_features: bool = False):
def create_item_embeddings_batch_feature_view(source, infer_features: bool = False) -> BatchFeatureView:
item_embeddings_feature_view = BatchFeatureView(
name="item_embeddings",
entities=["item"],
Expand All @@ -161,7 +161,7 @@ def create_item_embeddings_batch_feature_view(source, infer_features: bool = Fal
Field(name="embedding_double", dtype=Array(Float64)),
Field(name="embedding_float", dtype=Array(Float32)),
],
batch_source=source,
source=source,
ttl=timedelta(hours=2),
)
return item_embeddings_feature_view
Expand All @@ -184,7 +184,7 @@ def create_driver_hourly_stats_feature_view(source, infer_features: bool = False
return driver_stats_feature_view


def create_driver_hourly_stats_batch_feature_view(source, infer_features: bool = False):
def create_driver_hourly_stats_batch_feature_view(source, infer_features: bool = False) -> BatchFeatureView:
driver_stats_feature_view = BatchFeatureView(
name="driver_stats",
entities=["driver"],
Expand Down