fix typing complaints

dedupe/api.py

@@ -205,8 +205,18 @@ def partition(self, data, threshold=0.5):  # pragma: no cover
         _cleanup_scores(pair_scores)
         return clusters_eval
 
+    @overload
+    @staticmethod
+    def _add_singletons(all_ids: Iterable[int], clusters: ClustersInt) -> ClustersInt:
+        ...
+
+    @overload
+    @staticmethod
+    def _add_singletons(all_ids: Iterable[str], clusters: ClustersStr) -> ClustersStr:
+        ...
+
     @staticmethod
-    def _add_singletons(all_ids: Iterable[RecordID], clusters: Clusters) -> Clusters:
+    def _add_singletons(all_ids, clusters):
         singletons = set(all_ids)
 
         for record_ids, score in clusters:
@@ -997,9 +1007,19 @@ def search(
         else:
             return list(results)
 
+    @overload
     def _format_search_results(
-        self, search_d: Data, results: ArrayLinks
-    ) -> LookupResults:
+        self, search_d: DataInt, results: ArrayLinks
+    ) -> LookupResultsInt:
+        ...
+
+    @overload
+    def _format_search_results(
+        self, search_d: DataStr, results: ArrayLinks
+    ) -> LookupResultsStr:
+        ...
+
+    def _format_search_results(self, search_d, results):
         seen: set[RecordID] = set()
 
         for result in results:

dedupe/clustering.py

@@ -252,7 +252,7 @@ def cluster(
             for cluster in clusters.values():
                 if len(cluster) > 1:
                     scores = confidences(cluster, squared_distances, N)
-                    yield tuple(i_to_id[i] for i in cluster), scores
+                    yield tuple(i_to_id[i] for i in cluster), scores  # type: ignore[misc]
 
         else:
             ((ids, score),) = sub_graph