From 68aa7d7fab4c52b288d23451dff319113d718c56 Mon Sep 17 00:00:00 2001 From: Lorenzo Moreschini Date: Fri, 17 Mar 2023 17:59:09 +0100 Subject: [PATCH] Add support for record id of bytes type --- dedupe/core.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/dedupe/core.py b/dedupe/core.py index a73c7401..c88570f5 100644 --- a/dedupe/core.py +++ b/dedupe/core.py @@ -325,6 +325,8 @@ def sniff_id_type(ids: Sequence[tuple[RecordID, RecordID]]) -> RecordIDDType: elif python_type is int: int(example) # make sure we can cast to int dtype = int + elif python_type is bytes: + dtype = (bytes, 256) # type: ignore[assignment] else: raise ValueError("Invalid type for record id")