🐛 Fix deepcopy not copying metadata in __diot__

pwwang · Oct 20, 2023 · d7eef33 · d7eef33
1 parent 267f6a1
commit d7eef33
Show file tree

Hide file tree

Showing 2 changed files with 65 additions and 56 deletions.
diff --git a/diot/diot.py b/diot/diot.py
@@ -520,11 +520,10 @@ def copy(self) -> "Diot":
     __copy__ = copy
 
     def __deepcopy__(self, memo: Optional[Dict[int, Any]] = None) -> "Diot":
-        out = self.__class__(
-            diot_nest=self.__diot__["nest"],
-            diot_transform=self.__diot__["transform"],
-            diot_frozen=self.__diot__["frozen"],
-        )
+        out = self.__class__()
+        for dk, dv in self.__diot__.items():
+            out.__diot__[dk] = deepcopy(dv)
+
         memo = memo or {}
         memo[id(self)] = out
         for key, value in self.items():

diff --git a/poetry.lock b/poetry.lock