Fix entity row to use join key instead of name

Signed-off-by: Felix Wang <[email protected]>
feast-dev · Apr 11, 2022 · f591c47 · f591c47
1 parent eb095f3
commit f591c47
Showing 1 changed file with 1 addition and 1 deletion.
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)
     ]