revert entity columns

Signed-off-by: Oleksii Moskalenko <[email protected]>

sdk/python/feast/feature_logging.py

@@ -11,7 +11,7 @@
     FeatureViewNotFoundException,
     OnDemandFeatureViewNotFoundException,
 )
-from feast.feature_view import DUMMY_ENTITY_ID
+from feast.feature_view import DUMMY_ENTITY_NAME
 from feast.protos.feast.core.FeatureService_pb2 import (
     LoggingConfig as LoggingConfigProto,
 )
@@ -77,10 +77,12 @@ def get_schema(self, registry: "Registry") -> pa.Schema:
                         fields[field.name] = FEAST_TYPE_TO_ARROW_TYPE[field.dtype]
 
             else:
-                for entity_column in feature_view.entity_columns:
-                    if entity_column.name == DUMMY_ENTITY_ID:
+                for entity_name in feature_view.entities:
+                    if entity_name == DUMMY_ENTITY_NAME:
                         continue
 
+                    entity = registry.get_entity(entity_name, self._project)
+
                     join_key = projection.join_key_map.get(
                         entity.join_key, entity.join_key
                     )

sdk/python/feast/feature_view.py

@@ -285,6 +285,7 @@ def __copy__(self):
             online=self.online,
         )
         fv.projection = copy.copy(self.projection)
+        fv.entities = self.entities
         return fv
 
     def __eq__(self, other):

sdk/python/tests/integration/e2e/test_validation.py

@@ -16,7 +16,6 @@
 )
 from feast.protos.feast.serving.ServingService_pb2 import FieldStatus
 from feast.wait import wait_retry_backoff
-
 from tests.integration.feature_repos.repo_configuration import (
     construct_universal_feature_views,
 )
@@ -27,7 +26,6 @@
 )
 from tests.utils.logged_features import prepare_logs
 
-
 _features = [
     "customer_profile:current_balance",
     "customer_profile:avg_passenger_count",