fix hashability of frozen Resource dataclass (use FrozenSet instead o…

…f Set)
datahub-project · Aug 13, 2024 · 5d7b4fb · 5d7b4fb
1 parent 1347cd6
commit 5d7b4fb
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/metadata-ingestion/src/datahub/ingestion/source/sac/sac.py b/metadata-ingestion/src/datahub/ingestion/source/sac/sac.py
@@ -682,7 +682,7 @@ def get_resources(self) -> Iterable[Resource]:
                 open_url=entity.openURL,
                 ancestor_path=ancestor_path,
                 is_mobile=entity.isMobile,
-                resource_models=resource_models,
+                resource_models=frozenset(resource_models),
             )
 
     def get_import_data_model_ids(self) -> Set[str]:

diff --git a/metadata-ingestion/src/datahub/ingestion/source/sac/sac_common.py b/metadata-ingestion/src/datahub/ingestion/source/sac/sac_common.py
@@ -1,6 +1,6 @@
 from dataclasses import dataclass
 from datetime import datetime
-from typing import Optional, Set
+from typing import FrozenSet, Optional
 
 
 @dataclass(frozen=True)
@@ -30,7 +30,7 @@ class Resource:
     open_url: str
     ancestor_path: Optional[str]
     is_mobile: bool
-    resource_models: Set[ResourceModel]
+    resource_models: FrozenSet[ResourceModel]
 
 
 @dataclass(frozen=True)