__all__ needs to be strings

dedupeio · Dec 11, 2022 · d2db6b0 · d2db6b0
1 parent 0c7943e
commit d2db6b0
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 12 deletions.
diff --git a/dedupe/datamodel.py b/dedupe/datamodel.py
@@ -153,7 +153,10 @@ def typify_variables(
     variable_definitions: Iterable[VariableDefinition],
 ) -> tuple[list[FieldVariable], list[Variable]]:
 
-    variable_types = {variable.type: variable for variable in dedupe.variables.__all__}
+    variable_types = {}
+    for variablename in dedupe.variables.__all__:
+        variable = getattr(dedupe.variables, variablename)
+        variable_types[variable.type] = variable
 
     plugin_vars = load_setuptools_entrypoints("dedupevariables")
     for var in plugin_vars:

diff --git a/dedupe/variables/__init__.py b/dedupe/variables/__init__.py
@@ -10,15 +10,15 @@
 from dedupe.variables.string import ShortStringType, StringType, TextType
 
 __all__ = [
-    CustomType,
-    CategoricalType,
-    ExactType,
-    ExistsType,
-    InteractionType,
-    LatLongType,
-    PriceType,
-    SetType,
-    ShortStringType,
-    StringType,
-    TextType,
+    "CustomType",
+    "CategoricalType",
+    "ExactType",
+    "ExistsType",
+    "InteractionType",
+    "LatLongType",
+    "PriceType",
+    "SetType",
+    "ShortStringType",
+    "StringType",
+    "TextType",
 ]