Correctly restore data for container

python-poetry · May 20, 2021 · 60da0c0 · 60da0c0
1 parent 1b610fd
commit 60da0c0
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/tomlkit/container.py b/tomlkit/container.py
@@ -654,13 +654,18 @@ def __reduce_ex__(self, protocol):
         return (
             self.__class__,
             self._getstate(protocol),
-            (self._map, self._body, self._parsed),
+            (self._map, self._body, self._parsed, self._table_keys),
         )
 
     def __setstate__(self, state):
         self._map = state[0]
         self._body = state[1]
         self._parsed = state[2]
+        self._table_keys = state[3]
+
+        for key, item in self._body:
+            if key is not None:
+                dict.__setitem__(self, key.key, item.value)
 
     def copy(self):  # type: () -> Container
         return copy.copy(self)