Pass project from context to get entities (#1137)

Signed-off-by: Oleksii Moskalenko <moskalenko.alexey@gmail.com>
feast-dev · Nov 5, 2020 · 60f24c8 · 60f24c8
1 parent 9285bd8
commit 60f24c8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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,