Merge 1212a7b into c595052

.gitignore

@@ -1,5 +1,6 @@
 .env
 .coverage*
+htmlcov
 cpredicates.c
 *.code-workspace
 libdistance-0.2.1
@@ -29,7 +30,6 @@ mysql.cnf
 *.egg-info
 ENV
 dist
-.coverage
 src/*.c
 
 !benchmarks/asv.conf.json

dedupe/_typing.py

@@ -81,10 +81,14 @@ class TrainingData(TypedDict):
 
 
 class Classifier(Protocol):
-    def fit(self, X: object, y: object) -> None:
+    """Takes an array of pairwise distances and computes the likelihood they are a pair."""
+
+    def fit(self, X: numpy.typing.NDArray[numpy.float_], y: LabelsLike) -> None:
         ...
 
-    def predict_proba(self, X: object) -> numpy.typing.NDArray[numpy.float_]:
+    def predict_proba(
+        self, X: numpy.typing.NDArray[numpy.float_]
+    ) -> numpy.typing.NDArray[numpy.float_]:
         ...
 
 

dedupe/api.py

@@ -1278,8 +1278,6 @@ class Dedupe(ActiveMatching, DedupeMatching):
     entity.
     """
 
-    ActiveLearner = labeler.DedupeDisagreementLearner
-
     def prepare_training(
         self,
         data: Data,
@@ -1321,7 +1319,7 @@ def prepare_training(
         # existing training data, so add them to data dictionary
         examples, y = flatten_training(self.training_pairs)
 
-        self.active_learner = self.ActiveLearner(
+        self.active_learner = labeler.DedupeDisagreementLearner(
             self.data_model,
             data,
             index_include=examples,
@@ -1341,8 +1339,6 @@ class Link(ActiveMatching):
     Mixin Class for Active Learning Record Linkage
     """
 
-    ActiveLearner = labeler.RecordLinkDisagreementLearner
-
     def prepare_training(
         self,
         data_1: Data,
@@ -1390,7 +1386,7 @@ def prepare_training(
         # existing training data, so add them to data dictionaries
         examples, y = flatten_training(self.training_pairs)
 
-        self.active_learner = self.ActiveLearner(
+        self.active_learner = labeler.RecordLinkDisagreementLearner(
             self.data_model,
             data_1,
             data_2,