Fix use get with default

Cody Lin · Cody Lin · commit 44b4e4910e11 · 2021-10-11T12:05:59.000-07:00
Signed-off-by: Cody Lin &lt;codyl@twitter.com&gt;
diff --git a/sdk/python/feast/feature_store.py b/sdk/python/feast/feature_store.py
@@ -844,12 +844,15 @@ def get_online_features(
                 entity = self._registry.get_entity(
                     entity_name, self.project, allow_cache=True
                 )
-                join_key_substitute = feature_view.projection.join_key_map.get(
-                    entity.join_key, entity.name
-                )
                 # User directly uses join_key as the entity reference in the entity_rows for the
                 # entity mapping case.
-                entity_name_to_join_key_map[join_key_substitute] = join_key_substitute
+                entity_name = feature_view.projection.join_key_map.get(
+                    entity.join_key, entity.name
+                )
+                join_key = feature_view.projection.join_key_map.get(
+                    entity.join_key, entity.join_key
+                )
+                entity_name_to_join_key_map[entity_name] = join_key
 
         needed_request_data_features = self._get_needed_request_data_features(
             grouped_odfv_refs
