update datamodel version

dedupeio · Jun 21, 2024 · b5dece5 · b5dece5
1 parent 7ca4b6f
commit b5dece5
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/dedupe/datamodel.py b/dedupe/datamodel.py
@@ -24,7 +24,7 @@
 
 
 class DataModel(object):
-    version = 1
+    version = 2
 
     def __init__(self, variable_definitions: Iterable[Variable]):
         variable_definitions = list(variable_definitions)
@@ -151,6 +151,8 @@ def __setstate__(self, d):
         if version is None and "_variables" in d:
             d["_len"] = len(d.pop("_variables"))
             d["primary_variables"] = d.pop("primary_fields")
+        elif version == 1:
+            d["field_variables"] = d.pop("primary_variables")
 
         self.__dict__ = d