From 41b8b822a631cf6ef2130eded8535e3bbaf0bb52 Mon Sep 17 00:00:00 2001 From: Oleksii Moskalenko Date: Thu, 5 Nov 2020 11:19:18 +0800 Subject: [PATCH] pass project to get_entity Signed-off-by: Oleksii Moskalenko --- sdk/python/feast/pyspark/launcher.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/sdk/python/feast/pyspark/launcher.py b/sdk/python/feast/pyspark/launcher.py index e32a5663c3a..e3e7c03e51d 100644 --- a/sdk/python/feast/pyspark/launcher.py +++ b/sdk/python/feast/pyspark/launcher.py @@ -137,7 +137,7 @@ def _feature_table_to_argument( "project": project, "name": feature_table.name, "entities": [ - {"name": n, "type": client.get_entity(n).value_type} + {"name": n, "type": client.get_entity(n, project=project).value_type} for n in feature_table.entities ], "max_age": feature_table.max_age.ToSeconds() if feature_table.max_age else None,