consistent frozenset

dedupeio · Feb 17, 2023 · 296beb2 · 296beb2
1 parent dd67074
commit 296beb2
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/dedupe/labeler.py b/dedupe/labeler.py
@@ -153,7 +153,7 @@ def _predict(self, pairs: TrainingExamples) -> Labels:
                         "a frozen set, this will soon be required behaviour",
                         DeprecationWarning,
                     )
-                    overlap = not set(keys).isdisjoint(predicate(record_1))
+                    overlap = not frozenset(keys).isdisjoint(predicate(record_1))
                 if overlap:
                     labels.append(1)
                     break

diff --git a/dedupe/training.py b/dedupe/training.py
@@ -176,7 +176,7 @@ def cover(self, pairs: TrainingExamples, index_predicates: bool = True) -> Cover
                 coverage = frozenset(
                     i
                     for i, (record_1, record_2) in enumerate(pairs)
-                    if not set(predicate(record_1)).isdisjoint(
+                    if not frozenset(predicate(record_1)).isdisjoint(
                         predicate(record_2, target=True)
                     )
                 )