From f591c4704732827a2ca2dc373aaf9f45a3f1efcb Mon Sep 17 00:00:00 2001 From: Felix Wang Date: Mon, 11 Apr 2022 10:12:39 -0700 Subject: [PATCH] Fix entity row to use join key instead of name Signed-off-by: Felix Wang --- .../benchmarks/test_benchmark_universal_online_retrieval.py | 2 +- 1 file changed, 1 insertion(+), 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 index 3d2a253dc0..a29383a5c9 100644 --- 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) ]