fix: Fix entity row to use join key instead of name (feast-dev#2521)

felixwang9817 · web-flow · commit c22fa2c42c2c · 2022-04-11T10:45:56.000-07:00
Signed-off-by: Felix Wang &lt;wangfelix98@gmail.com&gt;
diff --git a/sdk/python/tests/benchmarks/test_benchmark_universal_online_retrieval.py b/sdk/python/tests/benchmarks/test_benchmark_universal_online_retrieval.py
@@ -38,7 +38,7 @@ def test_online_retrieval(environment, universal_data_sources, benchmark):
     sample_customers = random.sample(entities.customer_vals, 10)
 
     entity_rows = [
-        {"driver": d, "customer_id": c, "val_to_add": 50}
+        {"driver_id": d, "customer_id": c, "val_to_add": 50}
         for (d, c) in zip(sample_drivers, sample_customers)
     ]
 

Original file line number	Diff line number	Diff line change
`@@ -38,7 +38,7 @@ def test_online_retrieval(environment, universal_data_sources, benchmark):`
`38`	`38`	`sample_customers = random.sample(entities.customer_vals, 10)`
`39`	`39`
`40`	`40`	`entity_rows = [`
`41`		`- {"driver": d, "customer_id": c, "val_to_add": 50}`
	`41`	`+ {"driver_id": d, "customer_id": c, "val_to_add": 50}`
`42`	`42`	`for (d, c) in zip(sample_drivers, sample_customers)`
`43`	`43`	`]`
`44`	`44`