diff --git a/examples/java-demo/feature_repo/driver_repo.py b/examples/java-demo/feature_repo/driver_repo.py index 7eb98d2c21..cf00c74091 100644 --- a/examples/java-demo/feature_repo/driver_repo.py +++ b/examples/java-demo/feature_repo/driver_repo.py @@ -55,7 +55,7 @@ def transformed_conv_rate(inputs: pd.DataFrame) -> pd.DataFrame: # Define request feature view driver_age_request_fv = RequestFeatureView( name="driver_age", - request_source=RequestSource( + request_data_source=RequestSource( name="driver_age", schema={"driver_age": ValueType.INT64,} ), ) diff --git a/java/serving/src/test/resources/docker-compose/feast10/definitions.py b/java/serving/src/test/resources/docker-compose/feast10/definitions.py index 0b00704348..c48acffe17 100644 --- a/java/serving/src/test/resources/docker-compose/feast10/definitions.py +++ b/java/serving/src/test/resources/docker-compose/feast10/definitions.py @@ -7,14 +7,15 @@ from feast.feature_view import FeatureView from feast.entity import Entity from feast.feature_service import FeatureService -from feast.on_demand_feature_view import RequestDataSource, on_demand_feature_view +from feast.data_source import RequestSource +from feast.on_demand_feature_view import on_demand_feature_view from feast import FileSource file_path = "driver_stats.parquet" driver_hourly_stats = FileSource( path=file_path, - event_timestamp_column="event_timestamp", + timestamp_field="event_timestamp", created_timestamp_column="created", ) @@ -40,7 +41,7 @@ ) -input_request = RequestDataSource( +input_request = RequestSource( name="vals_to_add", schema={ "val_to_add": ValueType.INT64, @@ -68,7 +69,7 @@ def transformed_conv_rate(features_df: pd.DataFrame) -> pd.DataFrame: generated_data_source = FileSource( path="benchmark_data.parquet", - event_timestamp_column="event_timestamp", + timestamp_field="event_timestamp", ) entity = Entity(