fix: Fix __hash__ methods (#2556)

* Fix __hash__ method for Entity

Signed-off-by: Felix Wang <[email protected]>

* Fix __hash__ method for FeatureService

Signed-off-by: Felix Wang <[email protected]>

* Remove references to PrimitiveFeastType

Signed-off-by: Felix Wang <[email protected]>

* Fix __hash__ method for DataSource and PushSource

Signed-off-by: Felix Wang <[email protected]>

* Fix __hash__ method for FeatureView and OnDemandFeatureView

Signed-off-by: Felix Wang <[email protected]>

* Fix __hash__ method for SavedDataset

Signed-off-by: Felix Wang <[email protected]>

* Fix bugs

Signed-off-by: Felix Wang <[email protected]>

* Fix push merge

Signed-off-by: Danny Chiao <[email protected]>

Co-authored-by: Danny Chiao <[email protected]>

Author: felixwang9817

sdk/python/feast/base_feature_view.py

@@ -110,7 +110,7 @@ def __str__(self):
         return str(MessageToJson(self.to_proto()))
 
     def __hash__(self):
-        return hash((id(self), self.name))
+        return hash((self.name))
 
     def __getitem__(self, item):
         assert isinstance(item, list)
@@ -134,6 +134,7 @@ def __eq__(self, other):
         if (
             self.name != other.name
             or sorted(self.features) != sorted(other.features)
+            or self.projection != other.projection
             or self.description != other.description
             or self.tags != other.tags
             or self.owner != other.owner

sdk/python/feast/data_source.py

@@ -245,7 +245,7 @@ def __init__(
         self.owner = owner or ""
 
     def __hash__(self):
-        return hash((id(self), self.name))
+        return hash((self.name, self.timestamp_field))
 
     def __str__(self):
         return str(MessageToJson(self.to_proto()))
@@ -263,9 +263,9 @@ def __eq__(self, other):
             or self.created_timestamp_column != other.created_timestamp_column
             or self.field_mapping != other.field_mapping
             or self.date_partition_column != other.date_partition_column
+            or self.description != other.description
             or self.tags != other.tags
             or self.owner != other.owner
-            or self.description != other.description
         ):
             return False
 
@@ -392,6 +392,9 @@ def __eq__(self, other):
                 "Comparisons should only involve KafkaSource class objects."
             )
 
+        if not super().__eq__(other):
+            return False
+
         if (
             self.kafka_options.bootstrap_servers
             != other.kafka_options.bootstrap_servers
@@ -402,6 +405,9 @@ def __eq__(self, other):
 
         return True
 
+    def __hash__(self):
+        return super().__hash__()
+
     @staticmethod
     def from_proto(data_source: DataSourceProto):
         return KafkaSource(
@@ -507,13 +513,10 @@ def __eq__(self, other):
             raise TypeError(
                 "Comparisons should only involve RequestSource class objects."
             )
-        if (
-            self.name != other.name
-            or self.description != other.description
-            or self.owner != other.owner
-            or self.tags != other.tags
-        ):
+
+        if not super().__eq__(other):
             return False
+
         if isinstance(self.schema, List) and isinstance(other.schema, List):
             for field1, field2 in zip(self.schema, other.schema):
                 if field1 != field2:
@@ -671,24 +674,26 @@ def __init__(
         )
 
     def __eq__(self, other):
-        if other is None:
-            return False
-
         if not isinstance(other, KinesisSource):
             raise TypeError(
                 "Comparisons should only involve KinesisSource class objects."
             )
 
+        if not super().__eq__(other):
+            return False
+
         if (
-            self.name != other.name
-            or self.kinesis_options.record_format != other.kinesis_options.record_format
+            self.kinesis_options.record_format != other.kinesis_options.record_format
             or self.kinesis_options.region != other.kinesis_options.region
             or self.kinesis_options.stream_name != other.kinesis_options.stream_name
         ):
             return False
 
         return True
 
+    def __hash__(self):
+        return super().__hash__()
+
     def to_proto(self) -> DataSourceProto:
         data_source_proto = DataSourceProto(
             name=self.name,
@@ -744,6 +749,21 @@ def __init__(
         if not self.batch_source:
             raise ValueError(f"batch_source is needed for push source {self.name}")
 
+    def __eq__(self, other):
+        if not isinstance(other, PushSource):
+            raise TypeError("Comparisons should only involve PushSource class objects.")
+
+        if not super().__eq__(other):
+            return False
+
+        if self.batch_source != other.batch_source:
+            return False
+
+        return True
+
+    def __hash__(self):
+        return super().__hash__()
+
     def validate(self, config: RepoConfig):
         pass
 

sdk/python/feast/diff/registry_diff.py

@@ -177,7 +177,7 @@ def extract_objects_for_keep_delete_update_add(
         FeastObjectType, List[Any]
     ] = FeastObjectType.get_objects_from_registry(registry, current_project)
     registry_object_type_to_repo_contents: Dict[
-        FeastObjectType, Set[Any]
+        FeastObjectType, List[Any]
     ] = FeastObjectType.get_objects_from_repo_contents(desired_repo_contents)
 
     for object_type in FEAST_OBJECT_TYPES:

sdk/python/feast/entity.py

@@ -105,7 +105,7 @@ def __init__(
         self.last_updated_timestamp = None
 
     def __hash__(self) -> int:
-        return hash((id(self), self.name))
+        return hash((self.name, self.join_key))
 
     def __eq__(self, other):
         if not isinstance(other, Entity):

sdk/python/feast/feature_service.py

@@ -85,7 +85,7 @@ def __str__(self):
         return str(MessageToJson(self.to_proto()))
 
     def __hash__(self):
-        return hash((id(self), self.name))
+        return hash((self.name))
 
     def __eq__(self, other):
         if not isinstance(other, FeatureService):

sdk/python/feast/feature_store.py

@@ -533,25 +533,25 @@ def _plan(
             ...     batch_source=driver_hourly_stats,
             ... )
             >>> registry_diff, infra_diff, new_infra = fs._plan(RepoContents(
-            ...     data_sources={driver_hourly_stats},
-            ...     feature_views={driver_hourly_stats_view},
-            ...     on_demand_feature_views=set(),
-            ...     request_feature_views=set(),
-            ...     entities={driver},
-            ...     feature_services=set())) # register entity and feature view
+            ...     data_sources=[driver_hourly_stats],
+            ...     feature_views=[driver_hourly_stats_view],
+            ...     on_demand_feature_views=list(),
+            ...     request_feature_views=list(),
+            ...     entities=[driver],
+            ...     feature_services=list())) # register entity and feature view
         """
         # Validate and run inference on all the objects to be registered.
         self._validate_all_feature_views(
-            list(desired_repo_contents.feature_views),
-            list(desired_repo_contents.on_demand_feature_views),
-            list(desired_repo_contents.request_feature_views),
+            desired_repo_contents.feature_views,
+            desired_repo_contents.on_demand_feature_views,
+            desired_repo_contents.request_feature_views,
         )
-        _validate_data_sources(list(desired_repo_contents.data_sources))
+        _validate_data_sources(desired_repo_contents.data_sources)
         self._make_inferences(
-            list(desired_repo_contents.data_sources),
-            list(desired_repo_contents.entities),
-            list(desired_repo_contents.feature_views),
-            list(desired_repo_contents.on_demand_feature_views),
+            desired_repo_contents.data_sources,
+            desired_repo_contents.entities,
+            desired_repo_contents.feature_views,
+            desired_repo_contents.on_demand_feature_views,
         )
 
         # Compute the desired difference between the current objects in the registry and

sdk/python/feast/feature_view.py

@@ -270,7 +270,6 @@ def _initialize_sources(self, name, batch_source, stream_source, source):
                 self.batch_source = batch_source
         self.source = source
 
-    # Note: Python requires redefining hash in child classes that override __eq__
     def __hash__(self):
         return super().__hash__()
 
@@ -298,19 +297,15 @@ def __eq__(self, other):
             return False
 
         if (
-            self.tags != other.tags
+            sorted(self.entities) != sorted(other.entities)
             or self.ttl != other.ttl
             or self.online != other.online
+            or self.batch_source != other.batch_source
+            or self.stream_source != other.stream_source
+            or self.schema != other.schema
         ):
             return False
 
-        if sorted(self.entities) != sorted(other.entities):
-            return False
-        if self.batch_source != other.batch_source:
-            return False
-        if self.stream_source != other.stream_source:
-            return False
-
         return True
 
     def ensure_valid(self):

sdk/python/feast/on_demand_feature_view.py

@@ -234,14 +234,19 @@ def __copy__(self):
         return fv
 
     def __eq__(self, other):
+        if not isinstance(other, OnDemandFeatureView):
+            raise TypeError(
+                "Comparisons should only involve OnDemandFeatureView class objects."
+            )
+
         if not super().__eq__(other):
             return False
 
         if (
-            not self.source_feature_view_projections
-            == other.source_feature_view_projections
-            or not self.source_request_sources == other.source_request_sources
-            or not self.udf.__code__.co_code == other.udf.__code__.co_code
+            self.source_feature_view_projections
+            != other.source_feature_view_projections
+            or self.source_request_sources != other.source_request_sources
+            or self.udf.__code__.co_code != other.udf.__code__.co_code
         ):
             return False
 

sdk/python/feast/registry.py

@@ -18,7 +18,7 @@
 from enum import Enum
 from pathlib import Path
 from threading import Lock
-from typing import Any, Dict, List, Optional, Set
+from typing import Any, Dict, List, Optional
 from urllib.parse import urlparse
 
 import dill
@@ -98,7 +98,7 @@ def get_objects_from_registry(
     @staticmethod
     def get_objects_from_repo_contents(
         repo_contents: RepoContents,
-    ) -> Dict["FeastObjectType", Set[Any]]:
+    ) -> Dict["FeastObjectType", List[Any]]:
         return {
             FeastObjectType.DATA_SOURCE: repo_contents.data_sources,
             FeastObjectType.ENTITY: repo_contents.entities,

sdk/python/feast/repo_contents.py

@@ -11,7 +11,7 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
-from typing import NamedTuple, Set
+from typing import List, NamedTuple
 
 from feast.data_source import DataSource
 from feast.entity import Entity
@@ -27,12 +27,12 @@ class RepoContents(NamedTuple):
     Represents the objects in a Feast feature repo.
     """
 
-    data_sources: Set[DataSource]
-    feature_views: Set[FeatureView]
-    on_demand_feature_views: Set[OnDemandFeatureView]
-    request_feature_views: Set[RequestFeatureView]
-    entities: Set[Entity]
-    feature_services: Set[FeatureService]
+    data_sources: List[DataSource]
+    feature_views: List[FeatureView]
+    on_demand_feature_views: List[OnDemandFeatureView]
+    request_feature_views: List[RequestFeatureView]
+    entities: List[Entity]
+    feature_services: List[FeatureService]
 
     def to_registry_proto(self) -> RegistryProto:
         registry_proto = RegistryProto()