From ea8696445a7cd2efc46a145cf242d10ea97a5b1b Mon Sep 17 00:00:00 2001
From: Forest Gregg <fgregg@datamade.us>
Date: Fri, 21 Jun 2024 09:14:27 -0400
Subject: [PATCH] update datamodel version

---
 dedupe/datamodel.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/dedupe/datamodel.py b/dedupe/datamodel.py
index 341ee535..1c8dca94 100644
--- 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