fix: Fix Java Master (#2499)

* Fix Signed-off-by: Kevin Zhang <kzhang@tecton.ai> * Fix? Signed-off-by: Kevin Zhang <kzhang@tecton.ai> * Update sources Signed-off-by: Kevin Zhang <kzhang@tecton.ai> * Fix Signed-off-by: Kevin Zhang <kzhang@tecton.ai>
feast-dev · Apr 6, 2022 · e083458 · e083458
1 parent 6abae16
commit e083458
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 5 deletions.
diff --git 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
@@ -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(