From 296beb2c3120ff259c22965e323674bd337c065f Mon Sep 17 00:00:00 2001 From: Forest Gregg Date: Fri, 17 Feb 2023 11:33:49 -0500 Subject: [PATCH] consistent frozenset --- dedupe/labeler.py | 2 +- dedupe/training.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/dedupe/labeler.py b/dedupe/labeler.py index 042c180b..52e87cd7 100644 --- 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 index 77395a7b..b38d5c73 100644 --- 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) ) )